Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix_test.cc139 XLA_TEST_F(MatrixTest, ParseEinsumString) { in XLA_TEST_F() argument
160 ParseEinsumString(to_string(test_case[0], test_case[1], test_case[2])); in XLA_TEST_F()
174 auto parse_result_or_status = ParseEinsumString(test_case); in XLA_TEST_F()
181 auto parse_result_or_status = ParseEinsumString(test_case); in XLA_TEST_F()
Dmatrix.h85 StatusOr<std::array<std::vector<int64>, 3>> ParseEinsumString(
Dmatrix.cc282 StatusOr<std::array<std::vector<int64>, 3>> ParseEinsumString( in ParseEinsumString() function
333 ParseEinsumString(einsum_config)); in Einsum()