Searched refs:rhs_outer_dims (Results 1 – 3 of 3) sorted by relevance
442 absl::InlinedVector<int64_t, 8> rhs_outer_dims; in Einsum() local455 rhs_outer_dims.push_back(rhs_dim); in Einsum()466 rhs_outer_dims.push_back(rhs_dim); in Einsum()480 rhs_outer_dims.push_back(i); in Einsum()484 absl::c_sort(rhs_outer_dims); in Einsum()507 for (auto d : rhs_outer_dims) { in Einsum()
2699 int64_t rhs_outer_dims = in HandleDot() local2702 CHECK(lhs_outer_dims == 0 || rhs_outer_dims == 0); in HandleDot()2703 if (rhs_outer_dims > 0) { in HandleDot()2710 dnums.lhs_batch_dimensions_size() + rhs_outer_dims); in HandleDot()2733 const int64_t outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot()
1265 std::vector<int64_t> rhs_outer_dims = RandomDims(0, 2); in ChooseXlaDotArguments() local1276 a.rhs_dims.insert(a.rhs_dims.end(), rhs_outer_dims.begin(), in ChooseXlaDotArguments()1277 rhs_outer_dims.end()); in ChooseXlaDotArguments()