Searched refs:rhs_outer_dims (Results 1 – 2 of 2) sorted by relevance
409 absl::InlinedVector<int64, 8> rhs_outer_dims; in Einsum() local422 rhs_outer_dims.push_back(rhs_dim); in Einsum()433 rhs_outer_dims.push_back(rhs_dim); in Einsum()447 rhs_outer_dims.push_back(i); in Einsum()451 absl::c_sort(rhs_outer_dims); in Einsum()474 for (auto d : rhs_outer_dims) { in Einsum()
2403 int64 rhs_outer_dims = in HandleDot() local2407 CHECK(lhs_outer_dims == 0 || rhs_outer_dims == 0); in HandleDot()2408 if (rhs_outer_dims > 0) { in HandleDot()2417 rhs_outer_dims); in HandleDot()2443 const int64 outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot()