Home
last modified time | relevance | path

Searched refs:dynamic_shape (Results 1 – 25 of 58) sorted by relevance

123

/third_party/mindspore/tests/st/ops/gpu/
Dtest_dynamic_shape_op.py36 def dynamic_shape(np_type): function
99 dynamic_shape(np.int32)
105 dynamic_shape(np.float16)
111 dynamic_shape(np.float32)
117 dynamic_shape(np.bool)
/third_party/mindspore/config/
Dop_info.config153 … 10, "kernel_name": "abs", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
154 …el_name": "inplace_add_d", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
155 …el_name": "inplace_sub_d", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
156 …"kernel_name": "abs_grad", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
157 …10, "kernel_name": "acos", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
158 …kernel_name": "acos_grad", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
159 …0, "kernel_name": "acosh", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
160 …ernel_name": "acosh_grad", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
161 …dam_apply_one_with_decay", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
162 …pply_centered_rms_prop_d", "partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need…
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Ddropout_unify_mindir.cc130 CNodePtr dynamic_shape = func_graph->NewCNode(dynamic_shape_inputs); in CreateDynamicShapeCNode() local
131 MS_EXCEPTION_IF_NULL(dynamic_shape); in CreateDynamicShapeCNode()
139 dynamic_shape->set_abstract(dynamic_shape_abstract); in CreateDynamicShapeCNode()
140 return dynamic_shape; in CreateDynamicShapeCNode()
151 CNodePtr dynamic_shape = CreateDynamicShapeCNode(func_graph, dropout_input, input_shape); in CreateDropoutGenMaskCNode() local
152 dynamic_shape->set_scope(dropout->scope()); in CreateDropoutGenMaskCNode()
153 dropout_gen_mask_inputs.push_back(dynamic_shape); in CreateDropoutGenMaskCNode()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/
Dopinfo.h132 bool dynamic_shape() const { return dynamic_shape_; } in dynamic_shape() function
142 void set_dynamic_shape(bool dynamic_shape) { dynamic_shape_ = dynamic_shape; } in set_dynamic_shape() argument
/third_party/mindspore/mindspore/ops/_op_impl/tbe/
Dreciprocal_ds.py29 .dynamic_shape(True) \
Dconcat_ds.py26 .dynamic_shape(True) \
Dsoftmax_ds.py26 .dynamic_shape(True) \
Dsqrt_ds.py26 .dynamic_shape(True) \
Dmul_ds.py26 .dynamic_shape(True) \
Dunsorted_segment_sum_ds.py26 .dynamic_shape(True) \
Dexp_ds.py26 .dynamic_shape(True) \
Dlog_ds.py26 .dynamic_shape(True) \
Ddropout_do_mask_ds.py26 .dynamic_shape(True) \
Dreal_div_ds.py26 .dynamic_shape(True) \
Dsquare_ds.py29 .dynamic_shape(True) \
Dadd_n_ds.py26 .dynamic_shape(True) \
Dconv2d_backprop_filter_ds.py26 .dynamic_shape(True) \
Dneg_ds.py26 .dynamic_shape(True) \
Drelu_ds.py26 .dynamic_shape(True) \
Dminimum_ds.py27 .dynamic_shape(True)\
Dconv2d_ds.py26 .dynamic_shape(True) \
Dbatch_matmul_ds.py24 .dynamic_shape(True) \
Dlayer_norm_beta_gamma_backprop_v2_ds.py26 .dynamic_shape(True) \
Dsub_ds.py26 .dynamic_shape(True)\
/third_party/mindspore/mindspore/ops/
Dop_info_register.py487 def dynamic_shape(self, dynamic_shape=False): argument
494 self._is_bool(dynamic_shape)
495 self.dynamic_shape_ = dynamic_shape

123