Searched refs:EinsumDiagonalLabels (Results 1 – 1 of 1) sorted by relevance
239 absl::optional<std::array<std::vector<int64>, 3>> EinsumDiagonalLabels( in EinsumDiagonalLabels() function287 auto labels = EinsumDiagonalLabels(config); in EinsumDiagonal()303 auto labels = EinsumDiagonalLabels(config); in EinsumInverseDiagonal()358 auto x_diagonal_labels = EinsumDiagonalLabels(x_config); in Einsum()363 auto y_diagonal_labels = EinsumDiagonalLabels(y_config); in Einsum()368 auto output_diagonal_labels = EinsumDiagonalLabels(output_config); in Einsum()