Searched refs:EinsumDiagonalLabels (Results 1 – 1 of 1) sorted by relevance
270 std::optional<std::array<std::vector<int64_t>, 3>> EinsumDiagonalLabels( in EinsumDiagonalLabels() function318 auto labels = EinsumDiagonalLabels(config); in EinsumDiagonal()334 auto labels = EinsumDiagonalLabels(config); in EinsumInverseDiagonal()390 auto x_diagonal_labels = EinsumDiagonalLabels(x_config); in Einsum()395 auto y_diagonal_labels = EinsumDiagonalLabels(y_config); in Einsum()401 auto output_diagonal_labels = EinsumDiagonalLabels(output_config); in Einsum()