Searched refs:ExpectLayoutIs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 68 void ExpectLayoutIs(const Shape& shape, in ExpectLayoutIs() function in xla::__anona5bdbb650111::LayoutAssignmentTest
1160 ExpectLayoutIs(root->shape(), {3, 2, 0, 1}); in TEST_F()
1161 ExpectLayoutIs(root->operand(0)->shape(), {0, 2, 1}); in TEST_F()
1176 ExpectLayoutIs(root->shape(), {0, 2, 3, 1}); in TEST_F()
1177 ExpectLayoutIs(root->operand(0)->shape(), {0, 1, 2}); in TEST_F()
1210 ExpectLayoutIs(custom_call->shape(), {3, 2, 0, 1}); in TEST_F()
1211 ExpectLayoutIs(custom_call->operand(0)->shape(), {0, 1}); in TEST_F()
1212 ExpectLayoutIs(custom_call->operand(1)->shape(), {1, 0}); in TEST_F()
1236 ExpectLayoutIs(custom_call->shape(), {3, 2, 0, 1}); in TEST_F()
1263 ExpectLayoutIs(root->shape(), {2, 1, 0, 3}); in TEST_F()
[all …]
|