Searched refs:batch_dim_numbers (Results 1 – 1 of 1) sorted by relevance
1096 std::vector<int64> batch_dim_numbers(dim_numbers.lhs_batch_dimensions_size()); in ValidateDotDimensionNumbers() local1097 absl::c_iota(batch_dim_numbers, 0); in ValidateDotDimensionNumbers()1099 absl::c_equal(batch_dim_numbers, dim_numbers.lhs_batch_dimensions())); in ValidateDotDimensionNumbers()1101 absl::c_equal(batch_dim_numbers, dim_numbers.rhs_batch_dimensions())); in ValidateDotDimensionNumbers()