Searched refs:sign_change (Results 1 – 1 of 1) sorted by relevance
87 auto sign_change = xla::ConstantR2<float>( in BilinearWeights() local89 sign_change = xla::ConvertElementType(sign_change, xla_type); in BilinearWeights()92 xla::BroadcastInDim(sign_change, broadcast_dims, last_two_dims_indices); in BilinearWeights()