Home
last modified time | relevance | path

Searched defs:input_tensor_map (Results 1 – 25 of 43) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dtopkrouter_info.cc74 Shape input_tensor_map{2, -1, -1}; in InferTensorMap() local
Dtile_info.cc139 TensorMap input_tensor_map; in InferTensorMap() local
170 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Drange_info.cc53 TensorMap input_tensor_map = {}, output_tensor_map = {MAP_NONE}; in InferTensorMap() local
Dprelu_info.cc68 TensorMap input_tensor_map; in InferTensorMap() local
Dmaxpool_info.cc182 TensorMap input_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dgathernd_info.cc95 TensorMap input_tensor_map(inputs_shape_[0].size(), MAP_NONE); // the input can not split in InferTensorMap() local
Dslice_info.cc133 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
319 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Dscatter_ops_info.cc113 TensorMap input_tensor_map; in InferTensorMap() local
Davgpool_info.cc197 TensorMap input_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dpad_info.cc128 Shape input_tensor_map; in InferTensorMap() local
Drandom_distribute_info.cc81 TensorMap input_tensor_map; in InferTensorMap() local
Dconv3d_info.cc133 TensorMap input_tensor_map = {4, 3, 2, 1, -1}; in InferTensorMap() local
Dgroup_norm_info.cc91 TensorMap input_tensor_map(this->inputs_shape_[INPUT_INDEX].size(), MAP_NONE); in InferTensorMap() local
Dactivation_info.cc661 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
770 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
877 Shape input_tensor_map; in InferTensorMap() local
Dbatchnorm_info.cc164 TensorMap input_tensor_map; in InferTensorMap() local
Dreduce_method_info.cc207 Shape input_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps() local
259 Shape input_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps() local
Dgatherd_info.cc130 Shape input_tensor_map; in InferTensorMap() local
Dscatter_math_ops_info.cc93 TensorMap input_tensor_map, updates_tensor_map; in InferTensorMap() local
Dreduce_base_method_info.cc55 Shape input_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Ddebug_info_manager.cc438 …oManager::GetConstTensor(const std::map<std::string, mindspore::schema::Tensor *> &input_tensor_ma… in GetConstTensor()
462 …const std::map<std::string, mindspore::schema::Tensor *> &input_tensor_map, const quant::DebugMode… in GetOriginBeforeCallBack()
498 const std::map<std::string, mindspore::schema::Tensor *> &input_tensor_map, in GetQuantBeforeCallBack()
545 const std::map<std::string, mindspore::schema::Tensor *> &input_tensor_map, in GetBeforeCallBack()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/ops_info/
Dreshape_test.cc148 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
169 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dprelu_test.cc116 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
221 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dtanh_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dpow_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/mslite_bench/mslite_bench/tools/
Dcross_framework_accuracy.py131 input_tensor_map): argument

12