Searched refs:other_dim (Results 1 – 3 of 3) sorted by relevance
987 dim.merge_with(other_dim)988 for dim, other_dim in zip(self.dims, other.dims)
662 int64_t other_dim = operand_index == 0 ? dim.rhs : dim.lhs; in InferDotOperandSharding() local663 if (other_dim >= 0) { in InferDotOperandSharding()664 other_operand_dims_to_replicate.push_back(other_dim); in InferDotOperandSharding()
1442 const int32_t other_dim = c->Value(c->Dim(input_shape, input_rank - 1)); in MatrixDiagV2Shape() local1449 ", d_upper = ", upper_diag_index, " ", input_rank, " ", other_dim); in MatrixDiagV2Shape()