Searched defs:out_tensor_shape (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/ |
| D | reshape_layout_transfer_test.cc | 58 Arrangement out_tensor_shape; in InferUnifiedLayout() local
90 const TensorShape &out_tensor_shape, in InferUnifiedLayoutCheck()
114 const TensorShape &out_tensor_shape) { in ValidUnifiedLayoutCheck()
135 TensorShape out_tensor_shape = {128, 4096}; in TEST_F() local
150 TensorShape out_tensor_shape = {128, 4096}; in TEST_F() local
165 TensorShape out_tensor_shape = {128, 4096}; in TEST_F() local
174 TensorShape out_tensor_shape = {128, 4096}; in TEST_F() local
183 TensorShape out_tensor_shape = {2, 4}; in TEST_F() local
192 TensorShape out_tensor_shape = {2, 2, 8}; in TEST_F() local
201 TensorShape out_tensor_shape = {2, 2}; in TEST_F() local
[all …]
|
| D | tensor_redistribution_test.cc | 79 Arrangement out_tensor_shape; in TEST_F() local
132 Arrangement out_tensor_shape; in TEST_F() local
185 Arrangement out_tensor_shape; in TEST_F() local
|
| D | util_layout_gen_test.cc | 268 … const TensorMap &out_tensor_map, const TensorShape &out_tensor_shape) { in LayoutTransferValidLayoutChangeCheck()
319 const TensorMap &out_tensor_map, const TensorShape &out_tensor_shape) { in ValidLayoutChangeCheck()
|
| D | redistribution_layout_transfer_test.cc | 257 TensorShape out_tensor_shape = std::get<2>(layout_list[out]); in ValidRedistributionLayoutCheckAll() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/ |
| D | pass_utils.cc | 139 auto out_tensor_shape = out_tensor->shape(); in InsertPostTranspose() local
|