Searched refs:lhs_dim (Results 1 – 1 of 1) sorted by relevance
682 for (int64 lhs_dim : dimension_numbers.lhs_batch_dimensions()) { in InferDotOpShape() local683 dimensions.push_back(lhs.dimensions(lhs_dim)); in InferDotOpShape()684 is_dynamic.push_back(lhs.is_dynamic_dimension(lhs_dim)); in InferDotOpShape()