Searched refs:batch_dim_numbers (Results 1 – 1 of 1) sorted by relevance
900 std::vector<int64> batch_dim_numbers(dim_numbers.lhs_batch_dimensions_size()); in ValidateDotDimensionNumbers() local901 absl::c_iota(batch_dim_numbers, 0); in ValidateDotDimensionNumbers()903 absl::c_equal(batch_dim_numbers, dim_numbers.lhs_batch_dimensions())); in ValidateDotDimensionNumbers()905 absl::c_equal(batch_dim_numbers, dim_numbers.rhs_batch_dimensions())); in ValidateDotDimensionNumbers()