| /third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
| D | bn_training_reduce_grad.py | 27 .input(0, "grads", False, "required", "all", reshape_type="NC") \
28 .input(1, "x_norm", False, "required", "all", reshape_type="NC") \
34 .output(0, "y", False, "required", "all", reshape_type="NC") \
|
| D | space_to_batch_nd.py | 28 .input(0, "x", False, "required", "all", reshape_type="NH") \
29 .output(0, "y", False, "required", "all", reshape_type="NH") \
|
| D | batch_to_space_nd.py | 28 .input(0, "x", False, "required", "all", reshape_type="NH") \
29 .output(0, "y", False, "required", "all", reshape_type="NH") \
|
| D | bn_infer_grad.py | 27 .input(0, "grads", False, "required", "all", reshape_type="NC") \
30 .output(0, "x_backprop", False, "required", "all", reshape_type="NC") \
|
| D | bn_training_update_grad.py | 27 .input(0, "grads", False, "required", "all", reshape_type="NC") \
28 .input(1, "x", False, "required", "all", reshape_type="NC") \
|
| D | bn_infer.py | 27 .input(0, "x", False, "required", "all", reshape_type="NC") \
32 .output(0, "y", False, "required", "all", reshape_type="NC") \
|
| D | bn_training_update_v2.py | 27 .input(0, "x", False, "required", "all", reshape_type="NC") \
32 .output(0, "y", False, "required", "all", reshape_type="NC") \
|
| D | bn_training_update_v3.py | 27 .input(0, "x", False, "required", "all", reshape_type="NC") \
32 .output(0, "y", False, "required", "all", reshape_type="NC") \
|
| D | bn_training_update.py | 29 .input(0, "x", False, "required", "all", reshape_type="NC") \
36 .output(0, "y", False, "required", "all", reshape_type="NC") \
|
| D | dynamic_gru_v2.py | 37 .input(1, "weight_input", False, "required", "all", reshape_type="CN") \
38 .input(2, "weight_hidden", False, "required", "all", reshape_type="CN") \
|
| D | bn_training_reduce.py | 26 .input(0, "x", False, "required", "all", reshape_type="NC") \
|
| D | basic_lstm_cell_weight_grad.py | 29 .output(0, "dw", False, "required", "all", reshape_type="CN") \
|
| D | basic_lstm_cell_input_grad.py | 28 .input(1, "w", False, "required", "all", reshape_type="NC") \
|
| D | basic_lstm_cell.py | 33 .input(3, "w", False, "required", "all", reshape_type="CN") \
|
| D | dynamic_rnn.py | 39 .input(1, "w", False, "required", "all", reshape_type="CN") \
|
| /third_party/mindspore/config/ |
| D | op_info.config | 153 …so", "compute_cost": 10, "kernel_name": "abs", "partial_flag": true, "reshape_type": "", "dynamic_…
154 …ute_cost": 10, "kernel_name": "inplace_add_d", "partial_flag": true, "reshape_type": "", "dynamic_…
155 …ute_cost": 10, "kernel_name": "inplace_sub_d", "partial_flag": true, "reshape_type": "", "dynamic_…
156 …"compute_cost": 10, "kernel_name": "abs_grad", "partial_flag": true, "reshape_type": "", "dynamic_…
157 …o", "compute_cost": 10, "kernel_name": "acos", "partial_flag": true, "reshape_type": "", "dynamic_…
158 …compute_cost": 10, "kernel_name": "acos_grad", "partial_flag": true, "reshape_type": "", "dynamic_…
159 …", "compute_cost": 10, "kernel_name": "acosh", "partial_flag": true, "reshape_type": "", "dynamic_…
160 …ompute_cost": 10, "kernel_name": "acosh_grad", "partial_flag": true, "reshape_type": "", "dynamic_…
161 …0, "kernel_name": "adam_apply_one_with_decay", "partial_flag": true, "reshape_type": "", "dynamic_…
162 …0, "kernel_name": "apply_centered_rms_prop_d", "partial_flag": true, "reshape_type": "", "dynamic_…
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 337 auto reshape_type = io_info_item->reshape_type(); in GenBuilderItem() local
350 reshape_types->emplace_back(reshape_type); in GenBuilderItem()
362 reshape_types->emplace_back(reshape_type); in GenBuilderItem()
371 reshape_types->emplace_back(reshape_type); in GenBuilderItem()
397 op_io_info_new->set_reshape_type(op_io_info.reshape_type()); in CreateNewOpIOInfo()
578 op_io_info_new->set_reshape_type(op_io_info.reshape_type()); in CreateNewOpIOInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.h | 103 const AnfNodePtr &trans_data, const std::string &reshape_type = {""},
113 const std::string &reshape_type = std::string{});
|
| D | ascend_helper.cc | 244 … const AnfNodePtr &trans_data, const std::string &reshape_type, const TypeId &type_id) { in RefreshKernelBuildInfo() argument
251 builder->SetInputsReshapeType({reshape_type}); in RefreshKernelBuildInfo()
252 builder->SetOutputsReshapeType({reshape_type}); in RefreshKernelBuildInfo()
328 const std::string &reshape_type) { in AddCastOpNodeToGraph() argument
339 builder.SetInputsReshapeType({reshape_type}); in AddCastOpNodeToGraph()
340 builder.SetOutputsReshapeType({reshape_type}); in AddCastOpNodeToGraph()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/ |
| D | opinfo.h | 66 const std::string &reshape_type() const { return reshape_type_; } in reshape_type() function
76 void set_reshape_type(const std::string &reshape_type) { reshape_type_ = reshape_type; } in set_reshape_type() argument
|
| /third_party/mindspore/mindspore/ops/ |
| D | op_info_register.py | 476 def reshape_type(self, reshape_type): member in TBERegOp
483 self._is_string(reshape_type)
484 self.reshape_type_ = reshape_type
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | transforms.py | 125 reshape_type = {
132 return reshape_type[interp]
|