Searched defs:Einsum (Results 1 – 3 of 3) sorted by relevance
159 xla::XlaOp Einsum(xla::XlaOp x, absl::Span<const int64> x_config, xla::XlaOp y, in Einsum() function328 XlaOp Einsum(XlaOp x, XlaOp y, absl::string_view einsum_config, in Einsum() function
118 XLA_TEST_F(MatrixTest, Einsum) { in XLA_TEST_F() argument
1255 XLA_TEST_F(DotOperationTextTest, Einsum) { in XLA_TEST_F() argument