Home
last modified time | relevance | path

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

123

/third_party/mindspore/mindspore-src/source/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-src/source/mindspore/python/mindspore/parallel/
D_tensor.py28 def _get_tensor_strategy(dev_mat, tensor_map): argument
48 def _get_tensor_slice_index(device_arrangement, tensor_strategy, tensor_map, rank_index): argument
106 def _convert_to_new_device_coordinate(device_coordinate, tensor_map): argument
178 def _get_slice_index(dev_mat, tensor_map, opt_shard_group): argument
201 def _load_tensor(tensor, dev_mat, tensor_map, full_shape=None, rank_id=-1): argument
278 def _reshape_param_data(param_data, dev_mat, tensor_map): argument
389 def _construct_tensor_layout_for_opt_shard(dev_matrix, tensor_map, opt_shard_step, opt_shard_size, argument
Dshard.py91 def __call__(self, *tensor_map):
/third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/inner_headers/
Dmindir_memory_manager.h28 std::map<schema::Tensor *, TensorPtr> tensor_map; variable
/third_party/mindspore/mindspore-src/source/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-src/source/tests/ut/cpp/dataset/
Dbatch_op_test.cc131 TensorMap tensor_map; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.cc49 Status TensorLayout::Init(const Arrangement &device_arrangement, const Map &tensor_map, in Init()
72 Status TensorLayout::InitFromVector(const Shape &device_arrangement, const Shape &tensor_map, in InitFromVector()
111 …ensorLayout::InitFromExtendVector(const Shape &device_matrix, const std::vector<Shape> &tensor_map, in InitFromExtendVector()
633 Shape tensor_map(tensor_map_origin_.GetDimSize(), -1); in TransferRepeatLayout() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dvirtual_dataset_info.cc147 std::vector<ShapeBasePtr> tensor_map; in ObtainTensorMap() local
180 auto tensor_map = ObtainTensorMap(stra[i], slice_dim, dev_mat_origin); in InferTensorMapNew() local
Dinplace_op_info.cc62 Shape tensor_map = {-1}; in InferTensorMap() local
Dself_define_shard_info.cc119 for (const auto &tensor_map : outputs_tensor_map) { in InferAsLossDivisorByLayout() local
Drms_norm_info.cc103 Shape tensor_map; in CreateInputTensorMap() local
160 for (const auto &tensor_map : outputs_tensor_map) { in InferAsLossDivisorByLayout() local
Dslice_info.cc110 TensorMap tensor_map; in InferTensorMap() local
284 TensorMap tensor_map; in InferTensorMap() local
Dselect_info.cc100 TensorMap tensor_map; in InferTensorMap() local
Dconcat_info.cc118 TensorMap tensor_map; in InferTensorMap() local
Dactivation_info.cc187 int64_t tensor_map = inputs_tensor_map_[0][LongToSize(axis_index)]; in CheckLayoutConfig() local
356 auto tensor_map = tensor_layout.tensor_map_before(); in CheckInputLayout() local
475 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
597 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Dlayer_norm_info.cc176 Shape tensor_map; in CreateInputTensorMap() local
236 for (const auto &tensor_map : outputs_tensor_map) { in InferAsLossDivisorByLayout() local
Dtopk_info.cc115 TensorMap tensor_map; in InferTensorMap() local
Dsplit_info.cc178 TensorMap tensor_map; in InferTensorMap() local
Dgrid_sampler2d.cc95 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Dfillv2_info.cc62 TensorMap tensor_map; in InferTensorMap() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/
Dstrategy_checkpoint_info.cc45 auto tensor_map = layout_j.value().at("tensor_map").get<std::vector<int64_t>>(); in FromJson() local
157 straspb::TensorMap *tensor_map = parallel_layouts->add_tensor_map(); in to_protobuf() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dnode_strategy.proto53 repeated TensorMap tensor_map = 2; field
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/mslite_bench/mslite_bench/tools/
Dcross_framework_accuracy.py337 def check_np_dtype_with_model_input_dtype(tensor_map, argument

123