Searched defs:other_dim (Results 1 – 2 of 2) sorted by relevance
1070 int64 other_dim = operand_index == 0 ? dim.rhs : dim.lhs; in InferDotOperandSharding() local
1351 const int32 other_dim = c->Value(c->Dim(input_shape, input_rank - 1)); in MatrixDiagV2Shape() local