Searched refs:other_dim (Results 1 – 3 of 3) sorted by relevance
974 dim.merge_with(other_dim)975 for dim, other_dim in zip(self._dims, other.dims)
1070 int64 other_dim = operand_index == 0 ? dim.rhs : dim.lhs; in InferDotOperandSharding() local1071 if (other_dim >= 0) { in InferDotOperandSharding()1072 other_operand_dims_to_replicate.push_back(other_dim); in InferDotOperandSharding()
1351 const int32 other_dim = c->Value(c->Dim(input_shape, input_rank - 1)); in MatrixDiagV2Shape() local1358 ", d_upper = ", upper_diag_index, " ", input_rank, " ", other_dim); in MatrixDiagV2Shape()