Searched defs:out_tensor_map (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | reshape_layout_transfer_test.cc | 55 Map out_tensor_map; in InferUnifiedLayout() local
89 const TensorShape &in_tensor_shape, const TensorMap &out_tensor_map, in InferUnifiedLayoutCheck()
113 const TensorShape &in_tensor_shape, const TensorMap &out_tensor_map, in ValidUnifiedLayoutCheck()
134 TensorMap out_tensor_map = {1, 0}; in TEST_F() local
149 TensorMap out_tensor_map = {0, 1}; in TEST_F() local
164 TensorMap out_tensor_map = {0, 1}; in TEST_F() local
173 TensorMap out_tensor_map = {0, -1}; in TEST_F() local
182 TensorMap out_tensor_map = {-1, -1}; in TEST_F() local
191 TensorMap out_tensor_map = {-1, -1, 0}; in TEST_F() local
200 TensorMap out_tensor_map = {0, -1}; in TEST_F() local
[all …]
|
| D | redistribution_layout_transfer_test.cc | 55 Map out_tensor_map; in RedistributionLayoutTransferTestFunction() local
91 const TensorMap &out_tensor_map, in RedistributionLayoutCheck()
137 TensorMap out_tensor_map = {1, 0}; in TEST_F() local
172 TensorMap out_tensor_map = {0, 1}; in TEST_F() local
187 TensorMap out_tensor_map = {2, 1}; in TEST_F() local
202 TensorMap out_tensor_map = {2, 1}; in TEST_F() local
217 TensorMap out_tensor_map = {-1, 0}; in TEST_F() local
229 const TensorMap &out_tensor_map) { in ValidRedistributionLayoutCheck()
256 TensorMap out_tensor_map = std::get<1>(layout_list[out]); in ValidRedistributionLayoutCheckAll() local
|
| D | tensor_redistribution_test.cc | 76 Map out_tensor_map; in TEST_F() local
129 Map out_tensor_map; in TEST_F() local
182 Map out_tensor_map; in TEST_F() local
|
| D | redistribution_operator_infer_test.cc | 52 void InferOperatorCheck(Shape in_tensor_map, const Shape &out_tensor_map, const OperatorList &opera… in InferOperatorCheck()
91 Map out_tensor_map; in InferOperatorCheckAll() 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()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | pack_info.cc | 100 TensorMap out_tensor_map; in InferTensorMap() local
|
| D | broadcast_to_info.cc | 93 TensorMap out_tensor_map; in InferTensorMap() local
|
| D | unique_info.cc | 43 TensorMap out_tensor_map; in InferTensorMap() local
|
| D | gather_v2_p_info.cc | 551 Shape out_tensor_map = {1, -1, -1}; in InferTensorMap() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | redistribution_operator_infer.cc | 26 …tus RedistributionOperatorInfer::Init(const TensorLayout &tensor_layout, const Map &out_tensor_map, in Init()
|