Searched defs:lhs_dim (Results 1 – 5 of 5) sorted by relevance
257 for (int64 lhs_dim : dimension_numbers.lhs_batch_dimensions()) { in __anonbbc1c2d10202() local
107 const int lhs_dim = extended_lhs_shape.Dims(i); in ResizeOutputTensor() local366 const int lhs_dim = extended_lhs_shape.Dims(i); in Prepare() local
31 inline int broadcast_dim(int lhs_dim, int rhs_dim) { in broadcast_dim()
1277 int64_t lhs_dim = std::get<0>(dim); in GetCompatibleRankedTensorType() local
709 for (int64 lhs_dim : dimension_numbers.lhs_batch_dimensions()) { in InferDotOpShape() local