Home
last modified time | relevance | path

Searched refs:TensorMap (Results 1 – 25 of 67) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dreshape_layout_transfer_test.cc38 void InferUnifiedLayout(const DeviceArrangement &device_arrangement_shape, const TensorMap &in_tens… in InferUnifiedLayout()
39 … const TensorShape &in_tensor_shape_shape, const TensorMap &out_tensor_map_shape, in InferUnifiedLayout()
41 … TensorMap *unified_in_tensor_map_shape, TensorMap *unified_out_tensor_map_shape, in InferUnifiedLayout()
42 TensorMap *unified_tensor_shape_shape) { in InferUnifiedLayout()
88 void InferUnifiedLayoutCheck(const DeviceArrangement &device_arrangement, const TensorMap &in_tenso… in InferUnifiedLayoutCheck()
89 const TensorShape &in_tensor_shape, const TensorMap &out_tensor_map, in InferUnifiedLayoutCheck()
92 const TensorMap &unified_in_tensor_map_expect, in InferUnifiedLayoutCheck()
93 const TensorMap &unified_out_tensor_map_expect, in InferUnifiedLayoutCheck()
94 const TensorMap &unified_tensor_shape_expect) { in InferUnifiedLayoutCheck()
96 TensorMap unified_in_tensor_map; in InferUnifiedLayoutCheck()
[all …]
Dredistribution_layout_transfer_test.cc38 const DeviceArrangement &in_device_arrangement_shape, const TensorMap &in_tensor_map_shape, in RedistributionLayoutTransferTestFunction()
40 const TensorMap &out_tensor_map_shape, DeviceArrangement *unified_device_arrangement_shape, in RedistributionLayoutTransferTestFunction()
41 TensorMap *unified_in_tensor_map_shape, TensorMap *unified_out_tensor_map_shape, in RedistributionLayoutTransferTestFunction()
42 TensorMap *unified_tensor_shape_shape) { in RedistributionLayoutTransferTestFunction()
89 void RedistributionLayoutCheck(const DeviceArrangement &in_device_arrangement, const TensorMap &in_… in RedistributionLayoutCheck()
91 const TensorMap &out_tensor_map, in RedistributionLayoutCheck()
93 const TensorMap &unified_in_tensor_map_expect, in RedistributionLayoutCheck()
94 const TensorMap &unified_out_tensor_map_expect, in RedistributionLayoutCheck()
95 const TensorMap &unified_tensor_shape_expect) { in RedistributionLayoutCheck()
97 TensorMap unified_in_tensor_map; in RedistributionLayoutCheck()
[all …]
Dutil_layout_gen_test.h35 TensorMap GenerateTensorMap(const int64_t &map_size, const Shape &pos_index, const Shape &pos_value…
37 void GenerateValidTensorMap(const DeviceArrangement &device_arrangement, const TensorMap &tensor_sh…
38 std::vector<TensorMap> *tensor_map_list);
42 std::vector<std::tuple<DeviceArrangement, TensorMap, TensorShape>> *layout_list);
44 size_t ComputeNoneNumber(const TensorMap &tensor_map);
46 bool ShapeIsDividedByDevice(const DeviceArrangement &device_arrangement, const TensorMap &tensor_ma…
49 bool CheckLayoutValid(const DeviceArrangement &device_arrangement, const TensorMap &tensor_map,
52 void ComputeAccumDeviceTOAccumShapeMap(const DeviceArrangement &device_arrangement, const TensorMap…
57 … const TensorMap &in_tensor_map, const TensorShape &in_tensor_shape,
59 … const TensorMap &out_tensor_map, const TensorShape &out_tensor_shape);
[all …]
Dtensor_layout_test.cc36 const TensorMap &in_tensor_map_shape, in ReshapeExpandDeviceArrangementTestFunction()
39 const TensorMap &out_tensor_map_shape, in ReshapeExpandDeviceArrangementTestFunction()
75 TensorMap tensor_map = {1, 0}; in TEST_F()
78 TensorMap tensor_map_expect = {3, 2, 1, 0}; in TEST_F()
96 TensorMap tensor_map = {0, 1}; in TEST_F()
99 TensorMap tensor_map_expect = {1, 0, 3, 2}; in TEST_F()
116 TensorMap tensor_map = {1, -1}; in TEST_F()
119 TensorMap tensor_map_expect = {3, 2, -1}; in TEST_F()
137 TensorMap tensor_map = {0, 1}; in TEST_F()
140 TensorMap tensor_map_expect = {0, 2, 1}; in TEST_F()
[all …]
Dutil_layout_gen_test.cc95 TensorMap GenerateTensorMap(const int64_t &map_size, const Shape &pos_index, const Shape &pos_value… in GenerateTensorMap()
96 TensorMap tensor_map(map_size, -1); in GenerateTensorMap()
107 std::vector<TensorMap> *tensor_map_list) { in GenerateValidTensorMap()
119 TensorMap none_map(tensor_shape.size(), -1); in GenerateValidTensorMap()
142 TensorMap tensor_map = GenerateTensorMap(shape_size, pos_index[j], pos_value[k]); in GenerateValidTensorMap()
153 std::vector<std::tuple<DeviceArrangement, TensorMap, TensorShape>> *layout_list) { in GenerateValidLayoutByDeviceSizeAndTensorSize() argument
161 std::vector<TensorMap> tensor_map_list; in GenerateValidLayoutByDeviceSizeAndTensorSize()
176 bool CheckLayoutValid(const DeviceArrangement &device_arrangement, const TensorMap &tensor_map, in CheckLayoutValid()
188 size_t ComputeNoneNumber(const TensorMap &tensor_map) { in ComputeNoneNumber()
198 bool ShapeIsDividedByDevice(const DeviceArrangement &device_arrangement, const TensorMap &tensor_ma… in ShapeIsDividedByDevice()
[all …]
/third_party/mindspore/tests/ut/cpp/dataset/
Dimage_folder_op_test.cc94 TensorMap tensor_map; in TEST_F()
120 TensorMap tensor_map; in TEST_F()
150 TensorMap tensor_map; in TEST_F()
186 TensorMap tensor_map; in TEST_F()
219 TensorMap tensor_map; in TEST_F()
256 TensorMap tensor_map; in TEST_F()
285 TensorMap tensor_map; in TEST_F()
317 TensorMap tensor_map; in TEST_F()
346 TensorMap tensor_map; in TEST_F()
380 TensorMap tensor_map; in TEST_F()
[all …]
Dmanifest_op_test.cc74 TensorMap tensor_map; in TEST_F()
104 TensorMap tensor_map; in TEST_F()
134 TensorMap tensor_map; in TEST_F()
167 TensorMap tensor_map; in TEST_F()
196 TensorMap tensor_map; in TEST_F()
Dalbum_op_test.cc79 TensorMap tensor_map; in TEST_F()
105 TensorMap tensor_map; in TEST_F()
132 TensorMap tensor_map; in TEST_F()
161 TensorMap tensor_map; in TEST_F()
Dbatch_op_test.cc63 TensorMap tensor_map; in TEST_F()
111 TensorMap tensor_map; in TEST_F()
164 TensorMap tensor_map; in TEST_F()
217 TensorMap tensor_map; in TEST_F()
270 TensorMap tensor_map; in TEST_F()
375 TensorMap tensor_map; in TEST_F()
Dceleba_op_test.cc82 TensorMap tensor_map; in TEST_F()
129 TensorMap tensor_map; in TEST_F()
161 TensorMap tensor_map; in TEST_F()
Dcifar_op_test.cc99 TensorMap tensor_map; in TEST_F()
128 TensorMap tensor_map; in TEST_F()
154 TensorMap tensor_map; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dtensordot_info.h62 void InferTensorMapAxesInt(const TensorMap &tensor_map_index);
63 …void InferTensorMapAxesTuple(size_t size, const TensorMap &input_a_tensor_map, const TensorMap &te…
Dmatmul_dds_info.cc134 TensorMap input_tensor_map_q; in InferTensorMap()
143 TensorMap input_tensor_map_k; in InferTensorMap()
152 TensorMap input_tensor_map_local_mask; in InferTensorMap()
161 TensorMap input_tensor_map_global_mask; in InferTensorMap()
170 TensorMap output_tensor_map_local_prob; in InferTensorMap()
181 TensorMap output_tensor_map_global_prob; in InferTensorMap()
Ddsd_matmul_info.cc108 TensorMap input_tensor_map_w1; in InferTensorMap()
117 TensorMap input_tensor_map_w2; in InferTensorMap()
126 TensorMap input_tensor_map_v; in InferTensorMap()
135 TensorMap output_tensor_map; in InferTensorMap()
Dtensordot_info.cc214 void TensorDotInfo::InferTensorMapAxesInt(const TensorMap &tensor_map_index) { in InferTensorMapAxesInt()
217 TensorMap input_b_tensor_map, output_tensor_map; in InferTensorMapAxesInt()
232 void TensorDotInfo::InferTensorMapAxesTuple(size_t size, const TensorMap &input_a_tensor_map, in InferTensorMapAxesTuple()
233 const TensorMap &tensor_map_index) { in InferTensorMapAxesTuple()
237 TensorMap input_b_tensor_map, output_tensor_map, tmp_b_map_index; in InferTensorMapAxesTuple()
284 TensorMap tensor_map_index, input_a_tensor_map; in InferTensorMap()
Dgathernd_info.cc94 TensorMap input_tensor_map(inputs_shape_[0].size(), MAP_NONE); // the input can not split in InferTensorMap()
97 TensorMap indices_tensor_map; in InferTensorMap()
103 TensorMap output_tensor_map(outputs_shape_[0].size(), MAP_NONE); in InferTensorMap()
Darithmetic_info.cc113 TensorMap SetExpendTensorMap(const Shape &strategy, const Shape &dev_matrix_shape) { in SetExpendTensorMap()
114 TensorMap tensor_map_index; in SetExpendTensorMap()
125 TensorMap SetTensorMap(const Shape &strategy_expend, const Shape &dev_matrix_shape, const Shape &st… in SetTensorMap()
126 TensorMap expend_map = SetExpendTensorMap(strategy_expend, dev_matrix_shape); in SetTensorMap()
Dbatchnorm_info.cc111 TensorMap input_tensor_map; in InferTensorMap()
112 TensorMap in_other_tensor_map; in InferTensorMap()
147 TensorMap tmp_map; in InferForwardCommunication()
Dprelu_info.cc68 TensorMap input_tensor_map; in InferTensorMap()
74 TensorMap param_tensor_map; in InferTensorMap()
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dmatmul_info_test.cc174 TensorMap mat_a_expect = {4, 3, 2, 1}; in TEST_F()
175 TensorMap mat_b_expect = {4, 3, 1, 0}; in TEST_F()
176 TensorMap output_expect = {4, 3, 2, 0}; in TEST_F()
200 TensorMap mat_a_expect = {4, 3, 2, 1}; in TEST_F()
201 TensorMap mat_b_expect = {0, 1}; in TEST_F()
202 TensorMap output_expect = {4, 3, 2, 0}; in TEST_F()
226 TensorMap mat_a_expect = {2, 1}; in TEST_F()
227 TensorMap mat_b_expect = {4, 3, 0, 1}; in TEST_F()
228 TensorMap output_expect = {4, 3, 2, 0}; in TEST_F()
329 TensorMap mat_a_expect = {2, 1}; in TEST_F()
[all …]
Dprelu_test.cc108 TensorMap input_expect = {4, 3, 2, 1}; in TEST_F()
109 TensorMap param_expect = {2}; in TEST_F()
110 TensorMap output_expect = {4, 3, 2, 1}; in TEST_F()
213 TensorMap input_expect = {2, 1}; in TEST_F()
214 TensorMap param_expect = {0}; in TEST_F()
215 TensorMap output_expect = {2, 1}; in TEST_F()
Dsoftmax_entropy_loss_info_test.cc114 TensorMap input_expect = {3, 2, 1, 0}; in TEST_F()
115 TensorMap label_expect = {3, 2, 1, 0}; in TEST_F()
116 TensorMap output_0_expect = {3}; in TEST_F()
117 TensorMap output_1_expect = {3, 2, 1, 0}; in TEST_F()
Dreshape_test.cc142 TensorMap input_expect = {4, 3, 2, 1}; in TEST_F()
143 TensorMap output_expect = {-1, -1}; in TEST_F()
163 TensorMap input_expect = {3, 2, 1, 0}; in TEST_F()
164 TensorMap output_expect = {-1, -1}; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Ddataset_iterator.h33 using TensorMap = std::unordered_map<std::string, std::shared_ptr<Tensor>>; variable
67 Status GetNextAsMap(TensorMap *out_map);
/third_party/mindspore/mindspore/ccsrc/utils/
Dnode_strategy.proto39 message TensorMap { message
53 repeated TensorMap tensor_map = 2;

123