Home
last modified time | relevance | path

Searched refs:NormalizeEinsumString (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix_test.cc274 XLA_TEST_F(MatrixTest, NormalizeEinsumString) { in XLA_TEST_F() argument
275 EXPECT_EQ(NormalizeEinsumString("a,b->ab"), ""); in XLA_TEST_F()
276 EXPECT_EQ(NormalizeEinsumString("ba"), "ba->ab"); in XLA_TEST_F()
277 EXPECT_EQ(NormalizeEinsumString("ab,dc"), "ab,dc->abcd"); in XLA_TEST_F()
278 EXPECT_EQ(NormalizeEinsumString("a,b"), "a,b->ab"); in XLA_TEST_F()
279 EXPECT_EQ(NormalizeEinsumString("...ba,ca..."), "...ba,ca...->...bc"); in XLA_TEST_F()
Dmatrix.h113 std::string NormalizeEinsumString(absl::string_view einsum_config);
Dmatrix.cc649 std::string NormalizeEinsumString(absl::string_view einsum_config) { in NormalizeEinsumString() function
677 auto new_config = NormalizeEinsumString(einsum_config); in Einsum()