| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | reorder_ops.h | 29 std::vector<TypeId> inputs_type; member
|
| D | graph_kernel_helper.cc | 79 const std::vector<TypeId> &inputs_type, in BuildSelectKernelBuildInfo()
88 const std::vector<TypeId> &inputs_type, in BuildSelectKernelBuildInfo()
|
| D | convert_bfloat16.cc | 58 auto inputs_type = build_info->GetAllInputDeviceTypes(); in UpdateBuildInfoInputDataType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/ |
| D | dvm_kernel_mod.cc | 59 void DvmKernelMod::Initialize(const std::vector<TypeId> &inputs_type, const std::vector<TypeId> &ou… in Initialize()
175 void SingleDvmKernelMod::Initialize(const std::vector<TypeId> &inputs_type, const std::vector<TypeI… in Initialize()
227 void ParallelDvmKernelMod::Initialize(const std::vector<TypeId> &inputs_type, const std::vector<Typ… in Initialize()
|
| D | dvm_kernel_build.cc | 622 auto inputs_type = AnfAlgo::GetAllInputDeviceTypes(node_); in BuildKernel() local
686 auto inputs_type = AnfAlgo::GetAllInputDeviceTypes(node_); in BuildKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/ |
| D | host_kernel_metadata.cc | 36 std::vector<TypeId> inputs_type{}; in HostMetadataInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_kernel_metadata.cc | 63 std::vector<TypeId> inputs_type{}; in AicpuMetadataInfoForSpecialNodes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hccl_kernel_metadata.cc | 96 std::vector<TypeId> inputs_type{}; in HcclMetadataInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | kernel_info_setter.cc | 329 bool IsNeedProcessFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type) { in IsNeedProcessFormatInfo()
370 void UpdateKernelFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeId> &inputs_type, in UpdateKernelFormatInfo()
474 … const std::vector<TypeId> &inputs_type, in PrintUnsupportedTypeWarning()
689 std::vector<TypeId> inputs_type; in SetKernelInfoWithMsg() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | matmul_biasadd_fusion.cc | 37 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | adam_fusion.cc | 38 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | adam_weight_decay_fusion.cc | 38 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | print_reduce_fusion.cc | 36 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | add_rms_norm_fusion.cc | 37 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | add_cast_rms_norm_cast_fusion.cc | 46 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| D | add_layer_norm_fusion.cc | 69 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/ |
| D | print_value_type.cc | 33 std::vector<TypeId> inputs_type; in GenerateKernelBuildInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/ |
| D | insert_type_transform_op.cc | 219 std::vector<TypeId> inputs_type{tuple_unfold_abs->size(), builder->GetInputDeviceType(kIndex0)}; in CreateRealMakeTupleByTupleUnfoldInput() local
270 std::vector<TypeId> inputs_type; in SetKernelInfoForNewCNode() local
323 std::vector<TypeId> inputs_type = {type_id}; in SetKernelInfoForValueNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | debug_ops.py | 518 def infer_dtype(self, *inputs_type):
|
| D | _inner_ops.py | 1435 def infer_dtype(self, inputs_type, rois_type): argument
1576 def infer_dtype(self, *inputs_type):
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/common/ |
| D | device_common_test.h | 183 std::vector<TypeId> inputs_type; in SetKernelInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/ |
| D | cpu_device_context.cc | 345 std::vector<TypeId> inputs_type; in SetControlOpInfo() local
|