Searched refs:ExpectTupleLayoutIs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 75 void ExpectTupleLayoutIs( in ExpectTupleLayoutIs() function in xla::__anona66179c30111::LayoutAssignmentTest
1265 ExpectTupleLayoutIs(custom_call->operand(0)->shape(), {{1, 0}, {0, 1}}); in TEST_F()
1291 ExpectTupleLayoutIs(m->result_shape(), {{1, 0}, {1, 0}}); in TEST_F()
1294 ExpectTupleLayoutIs(custom_call->shape(), {{1, 0}, {0, 1}}); in TEST_F()
1383 ExpectTupleLayoutIs(crs->shape(), {{0, 1}, {1, 0}}); in TEST_F()
1420 ExpectTupleLayoutIs(alltoall->shape(), {{1, 0}, {1, 0}}); in TEST_F()
|