| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | cast.cc | 43 auto input_dtype = in_tensors_.front()->data_type(); in CheckSpecs() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | fill.cc | 34 auto input_dtype = input_args[kInputIndex0]->cast<abstract::AbstractTypePtr>(); in FillInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | unsorted_segment_sum_fission.cc | 116 auto input_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(origin_node, 0); in Process() local
|
| D | gather_v2_ds_fission.cc | 166 auto input_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(origin_node, 0); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | derelu_fusion.cc | 61 auto input_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(relu, 0); in CreateReluV2() local
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | lstm.py | 42 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
|
| D | embedding.py | 47 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
|
| D | image.py | 133 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
|
| D | math.py | 50 def _check_input_dtype(param_name, input_dtype, allow_dtypes, cls_name): argument
|
| D | rnns.py | 42 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _embedding_cache_ops.py | 135 def check_dtype(self, input_dtype, per_group_size_dtype, group_num_dtype): argument
|
| D | _inner_ops.py | 645 def check_dtype(self, input_dtype): argument
692 def infer_type(self, input_dtype): argument
|
| D | math_ops.py | 5379 def infer_dtype(self, input_dtype): argument
5418 def infer_dtype(self, input_dtype): argument
5461 def infer_dtype(self, input_dtype): argument
|
| D | sponge_ops.py | 3022 def infer_dtype(self, input_dtype): argument
3058 def infer_dtype(self, input_dtype): argument
|
| D | nn_ops.py | 3562 def infer_dtype(self, input_dtype): argument
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | layers.py | 168 def _check_input_dtype(input_dtype, param_name, allow_dtypes, cls_name): argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | kernel_select_cpu.cc | 165 TypeId input_dtype = kernel_attr->GetInputAttr(0).first; in ExpandKernelAttr() local
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | _constexpr_utils.py | 545 def get_np_eps(input_dtype): argument
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 2662 auto input_dtype = input_abs->BuildType()->ToString(); in GetArgsSpec() local
|