Searched refs:lhs_outer_dims (Results 1 – 2 of 2) sorted by relevance
410 absl::InlinedVector<int64, 8> lhs_outer_dims; in Einsum() local425 lhs_outer_dims.push_back(i); in Einsum()436 lhs_outer_dims.push_back(i); in Einsum()440 lhs_outer_dims.push_back(i); in Einsum()466 for (auto d : lhs_outer_dims) { in Einsum()
2399 int64 lhs_outer_dims = in HandleDot() local2407 CHECK(lhs_outer_dims == 0 || rhs_outer_dims == 0); in HandleDot()2420 } else if (lhs_outer_dims > 0) { in HandleDot()2429 lhs_outer_dims); in HandleDot()2443 const int64 outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot()