Searched refs:need_check_supported (Results 1 – 24 of 24) sorted by relevance
/third_party/mindspore/config/ |
D | op_info.config | 153 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 154 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 155 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 156 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 157 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 158 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 159 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 160 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 161 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… 162 …partial_flag": true, "reshape_type": "", "dynamic_shape": false, "need_check_supported": false, "i… [all …]
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/ |
D | opinfo.h | 112 need_check_supported_ = opinfo.need_check_supported(); in OpInfo() 135 bool need_check_supported() const { return need_check_supported_; } in need_check_supported() function 155 …void set_need_check_supported(bool need_check_supported) { need_check_supported_ = need_check_supp… in set_need_check_supported() argument
|
/third_party/mindspore/mindspore/ops/ |
D | op_info_register.py | 509 def need_check_supported(self, need_check_supported=False): argument 516 self._is_bool(need_check_supported) 517 self.need_check_supported_ = need_check_supported
|
/third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
D | resize_nearest_neighbor_grad.py | 26 .need_check_supported(True) \
|
D | resize_bilinear_grad.py | 26 .need_check_supported(True) \
|
D | max_pool_with_argmax.py | 32 .need_check_supported(True) \
|
D | resize_nearest_neighbor.py | 26 .need_check_supported(True) \
|
D | resize_bilinear.py | 26 .need_check_supported(True) \
|
D | top_k.py | 26 .need_check_supported(True) \
|
D | reduce_prod.py | 26 .need_check_supported(True) \
|
D | max_pool_grad_with_argmax.py | 33 .need_check_supported(True) \
|
D | batch_matmul.py | 28 .need_check_supported(True) \
|
D | transpose_d.py | 29 .need_check_supported(True) \
|
D | matmul.py | 26 .need_check_supported(True) \
|
D | transpose_ds.py | 30 .need_check_supported(True) \
|
D | unsorted_segment_max.py | 26 .need_check_supported(True) \
|
D | unsorted_segment_prod.py | 26 .need_check_supported(True) \
|
D | unsorted_segment_min.py | 26 .need_check_supported(True) \
|
D | fill.py | 26 .need_check_supported(True) \
|
D | cast_ds.py | 26 .need_check_supported(True) \
|
D | cast.py | 27 .need_check_supported(True) \
|
D | pack.py | 26 .need_check_supported(False) \
|
D | transpose.py | 30 .need_check_supported(True) \
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
D | tbe_kernel_select.cc | 202 if (op_info.need_check_supported()) { in FilterInVaildKernelInfo()
|