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::__anona5bdbb650111::LayoutAssignmentTest
1271 ExpectTupleLayoutIs(custom_call->operand(0)->shape(), {{1, 0}, {0, 1}}); in TEST_F()
1297 ExpectTupleLayoutIs(m->result_shape(), {{1, 0}, {1, 0}}); in TEST_F()
1300 ExpectTupleLayoutIs(custom_call->shape(), {{1, 0}, {0, 1}}); in TEST_F()
|