Searched refs:rhs_outer_dims (Results 1 – 2 of 2) sorted by relevance
219 std::vector<int64> rhs_outer_dims; in Einsum() local223 rhs_outer_dims.push_back(i); in Einsum()239 for (auto d : rhs_outer_dims) { in Einsum()
1719 int64 rhs_outer_dims = in HandleDot() local1723 CHECK(lhs_outer_dims == 0 || rhs_outer_dims == 0); in HandleDot()1724 if (rhs_outer_dims > 0) { in HandleDot()1733 rhs_outer_dims); in HandleDot()1755 const int64 outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot()