Searched refs:lhs_outer_dims (Results 1 – 2 of 2) sorted by relevance
196 std::vector<int64> lhs_outer_dims; in Einsum() local215 lhs_outer_dims.push_back(i); in Einsum()236 for (auto d : lhs_outer_dims) { in Einsum()
1715 int64 lhs_outer_dims = in HandleDot() local1723 CHECK(lhs_outer_dims == 0 || rhs_outer_dims == 0); in HandleDot()1736 } else if (lhs_outer_dims > 0) { in HandleDot()1745 lhs_outer_dims); in HandleDot()1755 const int64 outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot()