Searched refs:ExpectTupleLayoutIs (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment_test.cc | 76 void ExpectTupleLayoutIs( in ExpectTupleLayoutIs() function in xla::__anoned7fa0720111::LayoutAssignmentTest
1301 ExpectTupleLayoutIs(custom_call->operand(0)->shape(), {{1, 0}, {0, 1}}); in TEST_F()
1327 ExpectTupleLayoutIs(m->result_shape(), {{1, 0}, {1, 0}}); in TEST_F()
1330 ExpectTupleLayoutIs(custom_call->shape(), {{1, 0}, {0, 1}}); in TEST_F()
1418 ExpectTupleLayoutIs(crs->shape(), {{0, 1}, {1, 0}}); in TEST_F()
1455 ExpectTupleLayoutIs(alltoall->shape(), {{1, 0}, {1, 0}}); in TEST_F()
|