Searched refs:ParseEinsumString (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix_test.cc | 219 XLA_TEST_F(MatrixTest, ParseEinsumString) { in XLA_TEST_F() argument
256 ParseEinsumString(to_string(test_case[0], test_case[1], test_case[2]), in XLA_TEST_F()
269 auto parse_result_or_status = ParseEinsumString(test_case, 3, 3); in XLA_TEST_F()
|
| D | matrix.h | 107 StatusOr<std::array<std::vector<int64>, 3>> ParseEinsumString(
|
| D | matrix.cc | 558 StatusOr<std::array<std::vector<int64>, 3>> ParseEinsumString( in ParseEinsumString() function
685 ParseEinsumString(einsum_config, x_shape.rank(), y_shape.rank())); in Einsum()
|