Home
last modified time | relevance | path

Searched defs:out_tensor_map (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dreshape_layout_transfer_test.cc55 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 …]
Dredistribution_layout_transfer_test.cc55 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
Dtensor_redistribution_test.cc76 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
Dredistribution_operator_infer_test.cc52 void InferOperatorCheck(Shape in_tensor_map, const Shape &out_tensor_map, const OperatorList &opera… in InferOperatorCheck()
91 Map out_tensor_map; in InferOperatorCheckAll() local
Dutil_layout_gen_test.cc268const 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/
Dpack_info.cc100 TensorMap out_tensor_map; in InferTensorMap() local
Dbroadcast_to_info.cc93 TensorMap out_tensor_map; in InferTensorMap() local
Dunique_info.cc43 TensorMap out_tensor_map; in InferTensorMap() local
Dgather_v2_p_info.cc551 Shape out_tensor_map = {1, -1, -1}; in InferTensorMap() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.cc26 …tus RedistributionOperatorInfer::Init(const TensorLayout &tensor_layout, const Map &out_tensor_map, in Init()