Searched refs:unique_labels (Results 1 – 1 of 1) sorted by relevance
241 std::vector<int64> unique_labels; in EinsumDiagonalLabels() local248 unique_labels.push_back(*label); in EinsumDiagonalLabels()254 if (unique_labels.size() == config.size()) { in EinsumDiagonalLabels()257 return {{unique_labels, reduce_dims, broadcast_dims}}; in EinsumDiagonalLabels()