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::__anona66179c30111::LayoutAssignmentTest
1154 ExpectLayoutIs(root->shape(), {3, 2, 0, 1}); in TEST_F()
1155 ExpectLayoutIs(root->operand(0)->shape(), {0, 2, 1}); in TEST_F()
1170 ExpectLayoutIs(root->shape(), {0, 2, 3, 1}); in TEST_F()
1171 ExpectLayoutIs(root->operand(0)->shape(), {0, 1, 2}); in TEST_F()
1204 ExpectLayoutIs(custom_call->shape(), {3, 2, 0, 1}); in TEST_F()
1205 ExpectLayoutIs(custom_call->operand(0)->shape(), {0, 1}); in TEST_F()
1206 ExpectLayoutIs(custom_call->operand(1)->shape(), {1, 0}); in TEST_F()
1230 ExpectLayoutIs(custom_call->shape(), {3, 2, 0, 1}); in TEST_F()
1257 ExpectLayoutIs(root->shape(), {2, 1, 0, 3}); in TEST_F()
[all …]
|