Searched refs:EinsumDiagonal (Results 1 – 1 of 1) sorted by relevance
284 xla::XlaOp EinsumDiagonal(XlaOp x, absl::Span<const int64> config) { in EinsumDiagonal() function360 return Einsum(EinsumDiagonal(x, x_config), x_diagonal_labels->at(0), y, in Einsum()365 return Einsum(x, x_config, EinsumDiagonal(y, y_config), in Einsum()