Searched refs:canonical_batch_dims (Results 1 – 1 of 1) sorted by relevance
312 std::vector<int64> canonical_batch_dims( in Run() local314 absl::c_iota(canonical_batch_dims, 0); in Run()315 if (!absl::c_equal(dnums.lhs_batch_dimensions(), canonical_batch_dims) || in Run()316 !absl::c_equal(dnums.rhs_batch_dimensions(), canonical_batch_dims)) { in Run()