Home
last modified time | relevance | path

Searched defs:tensor_map (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dtensor_layout_test.cc44 Map tensor_map; in ReshapeExpandDeviceArrangementTestFunction() local
75 TensorMap tensor_map = {1, 0}; in TEST_F() local
96 TensorMap tensor_map = {0, 1}; in TEST_F() local
116 TensorMap tensor_map = {1, -1}; in TEST_F() local
137 TensorMap tensor_map = {0, 1}; in TEST_F() local
148 TensorMap tensor_map = {1, -1, 0}; in TEST_F() local
164 Map tensor_map; in ExpandTensorShapeTestFunction() local
196 TensorMap tensor_map = {1, 0}; in TEST_F() local
207 TensorMap tensor_map = {1, 0}; in TEST_F() local
222 Map tensor_map; in TEST_F() local
[all …]
Dutil_layout_gen_test.cc96 TensorMap tensor_map(map_size, -1); in GenerateTensorMap() local
142 TensorMap tensor_map = GenerateTensorMap(shape_size, pos_index[j], pos_value[k]); in GenerateValidTensorMap() local
176 bool CheckLayoutValid(const DeviceArrangement &device_arrangement, const TensorMap &tensor_map, in CheckLayoutValid()
188 size_t ComputeNoneNumber(const TensorMap &tensor_map) { in ComputeNoneNumber()
198 …ol ShapeIsDividedByDevice(const DeviceArrangement &device_arrangement, const TensorMap &tensor_map, in ShapeIsDividedByDevice()
236 …ccumDeviceTOAccumShapeMap(const DeviceArrangement &device_arrangement, const TensorMap &tensor_map, in ComputeAccumDeviceTOAccumShapeMap()
Dtensor_redistribution_test.cc54 TensorMap tensor_map = {2, 0}; in TEST_F() local
107 TensorMap tensor_map = {2, 0}; in TEST_F() local
159 TensorMap tensor_map = {0, -1, -1, -1}; in TEST_F() local
/third_party/mindspore/mindspore/parallel/
D_tensor.py22 def _get_tensor_strategy(dev_mat, tensor_map): argument
42 def _get_tensor_slice_index(device_arrangement, tensor_strategy, tensor_map, rank_index): argument
100 def _convert_to_new_device_coordinate(device_coordinate, tensor_map): argument
172 def _get_slice_index(dev_mat, tensor_map): argument
189 def _load_tensor(tensor, dev_mat, tensor_map): argument
252 def _reshape_param_data(param_data, dev_mat, tensor_map): argument
/third_party/mindspore/tests/ut/cpp/dataset/
Dimage_folder_op_test.cc94 TensorMap tensor_map; in TEST_F() local
120 TensorMap tensor_map; in TEST_F() local
150 TensorMap tensor_map; in TEST_F() local
186 TensorMap tensor_map; in TEST_F() local
219 TensorMap tensor_map; in TEST_F() local
256 TensorMap tensor_map; in TEST_F() local
285 TensorMap tensor_map; in TEST_F() local
317 TensorMap tensor_map; in TEST_F() local
346 TensorMap tensor_map; in TEST_F() local
380 TensorMap tensor_map; in TEST_F() local
[all …]
Dbatch_op_test.cc63 TensorMap tensor_map; in TEST_F() local
111 TensorMap tensor_map; in TEST_F() local
164 TensorMap tensor_map; in TEST_F() local
217 TensorMap tensor_map; in TEST_F() local
270 TensorMap tensor_map; in TEST_F() local
375 TensorMap tensor_map; in TEST_F() local
Dmanifest_op_test.cc74 TensorMap tensor_map; in TEST_F() local
104 TensorMap tensor_map; in TEST_F() local
134 TensorMap tensor_map; in TEST_F() local
167 TensorMap tensor_map; in TEST_F() local
196 TensorMap tensor_map; in TEST_F() local
Dalbum_op_test.cc79 TensorMap tensor_map; in TEST_F() local
105 TensorMap tensor_map; in TEST_F() local
132 TensorMap tensor_map; in TEST_F() local
161 TensorMap tensor_map; in TEST_F() local
Dcifar_op_test.cc99 TensorMap tensor_map; in TEST_F() local
128 TensorMap tensor_map; in TEST_F() local
154 TensorMap tensor_map; in TEST_F() local
Dceleba_op_test.cc82 TensorMap tensor_map; in TEST_F() local
129 TensorMap tensor_map; in TEST_F() local
161 TensorMap tensor_map; in TEST_F() local
Dmap_op_test.cc176 TensorMap tensor_map; in TEST_F() local
224 TensorMap tensor_map; in TEST_F() local
286 TensorMap tensor_map; in TEST_F() local
364 TensorMap tensor_map; in TEST_F() local
625 TensorMap tensor_map; in TEST_F() local
725 TensorMap tensor_map; in TEST_F() local
Dir_callback_test.cc171 TensorMap tensor_map; in TEST_F() local
233 TensorMap tensor_map; in TEST_F() local
303 TensorMap tensor_map; in TEST_F() local
/third_party/mindspore/mindspore/lite/mindir/inner_headers/
Dmindir_memory_manager.h28 std::map<schema::Tensor *, TensorPtr> tensor_map; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.cc48 Status TensorLayout::Init(const Arrangement &device_arrangement, const Map &tensor_map, in Init()
71 Status TensorLayout::InitFromVector(const Shape &device_arrangement, const Shape &tensor_map, in InitFromVector()
410 Shape tensor_map(tensor_map_origin_.GetDimSize(), -1); in TransferRepeatLayout() local
421 Shape tensor_map = tensor_map_.array(); in GenerateOptShardSliceShape() local
/third_party/mindspore/tests/ut/cpp/parallel/
Ddevice_matrix_test.cc88 Shape tensor_map = {-1, 0}; in TEST_F() local
99 Shape tensor_map = {1, 0}; in TEST_F() local
110 Shape tensor_map = {-1, 0}; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dselect_info.cc75 TensorMap tensor_map; in InferTensorMap() local
Dtopk_info.cc65 TensorMap tensor_map; in InferTensorMap() local
Dsplit_info.cc116 TensorMap tensor_map; in InferTensorMap() local
Dconcat_info.cc121 TensorMap tensor_map; in InferTensorMap() local
Dstrided_slice_info.cc160 TensorMap tensor_map; in InferTensorMap() local
Dactivation_info.cc219 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
359 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Dlayer_norm_info.cc129 Shape tensor_map; in CreateInputTensorMap() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dget_parallel_info.cc43 auto tensor_map = tensor_layout->tensor_map().array(); in GetParameterLayout() local
/third_party/mindspore/mindspore/ccsrc/utils/
Dnode_strategy.proto53 repeated TensorMap tensor_map = 2; field
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc204 Shape tensor_map = tensor_info.tensor_layout().tensor_map().array(); in GetGroupByTensorInfo() local
384 auto tensor_map = tensor_layout->tensor_map().array(); in IsFullySplitParameter() local
584 auto tensor_map = tensor_layout->tensor_map().array(); in HandleAdaFactorOpt() local

12