| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | arithmetic_logic_cpu_kernel.h | 34 …explicit ArithmeticLogicCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in ArithmeticLogicCpuKernelMod()
59 …explicit ArithmeticComplexLogicCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_… in ArithmeticComplexLogicCpuKernelMod()
|
| D | fused_ada_factor_cpu_kernel.h | 34 explicit FusedAdaFactorCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in FusedAdaFactorCpuKernelMod()
|
| D | reduce_cpu_kernel.h | 33 explicit ReduceCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in ReduceCpuKernelMod()
|
| D | fast_gelu_grad_cpu_kernel.h | 34 explicit FastGeLUGradCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in FastGeLUGradCpuKernelMod()
|
| D | fast_gelu_cpu_kernel.h | 34 explicit FastGeLUCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in FastGeLUCpuKernelMod()
|
| D | in_place_op_v2_cpu_kernel.h | 35 explicit InPlaceOpV2CpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in InPlaceOpV2CpuKernelMod()
|
| D | in_place_op_cpu_kernel.h | 33 explicit InPlaceOpCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in InPlaceOpCpuKernelMod()
|
| D | relu_v3_cpu_kernel.h | 34 explicit ReLUV3CpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in ReLUV3CpuKernelMod()
|
| D | hsigmoid_grad_cpu_kernel.h | 34 explicit HSigmoidGradCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in HSigmoidGradCpuKernelMod()
|
| D | hsigmoid_cpu_kernel.h | 34 explicit HSigmoidCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in HSigmoidCpuKernelMod()
|
| D | matmul_cpu_kernel.h | 34 explicit MatMulCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in MatMulCpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | lite_kernel.cc | 86 int LiteKernel::UpdateThreadNumProcess(int32_t kernel_type, int64_t per_unit_load_num, int64_t per_… in UpdateThreadNumProcess()
93 int LiteKernel::UpdateThreadNumPass(int32_t kernel_type, int64_t per_unit_load_num, int64_t per_uni… in UpdateThreadNumPass()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | gcd_lcm_gpu_kernel.h | 36 explicit GcdLcmGpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in GcdLcmGpuKernelMod()
|
| D | bessel_gpu_kernel.h | 32 explicit BesselGpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in BesselGpuKernelMod()
|
| D | segment_ops_gpu_kernel.h | 32 explicit SegmentOpsGpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in SegmentOpsGpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | kernel_info_setter.cc | 76 kernel::OpImplyType GetImplyType(KernelType kernel_type) { in GetImplyType()
120 std::string GetSupportedTypesStr(const CNodePtr &kernel_node, KernelType kernel_type) { in GetSupportedTypesStr()
217 KernelType *kernel_type) { in SelectCustomKernel()
476 KernelType kernel_type) { in PrintUnsupportedTypeWarning()
559 KernelType *kernel_type, in GetSelectKernelResult()
611 … KernelType kernel_type) { in GetSelectKernelObjectTypeResult()
674 …:string, ExceptionType> SetKernelInfoWithMsg(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfoWithMsg()
751 void GPUGraphKernelInfo::SetKernelInfo(const CNodePtr &kernel_node, KernelType kernel_type) { in SetKernelInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/sequence/ |
| D | tensor_to_sequence_cpu_kernel.h | 31 explicit TensorToSeqCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in TensorToSeqCpuKernelMod()
|
| D | sequence_add_cpu_kernel.h | 33 explicit SequenceAddCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in SequenceAddCpuKernelMod()
|
| D | sequence_slice_grad_cpu_kernel.h | 33 …explicit SequenceSliceGradCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type)… in SequenceSliceGradCpuKernelMod()
|
| D | sequence_add_offset_cpu_kernel.h | 33 …explicit SequenceAddOffsetCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type)… in SequenceAddOffsetCpuKernelMod()
|
| D | sequence_slice_cpu_kernel.h | 33 explicit SequenceSliceCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in SequenceSliceCpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | memcpy_gpu_kernel.h | 32 explicit MemcpyGpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in MemcpyGpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/ |
| D | conv3d_transpose_cpu_kernel.h | 32 …explicit Conv3DTransposeCpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in Conv3DTransposeCpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse_grad/ |
| D | sparse_add_grad_gpu_kernel.h | 38 explicit SparseAddGradGpuKernelMod(const std::string &kernel_type) : kernel_type_(kernel_type) {} in SparseAddGradGpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | kernel_build_info.h | 62 KernelType kernel_type() const { return kernel_type_; } in kernel_type() function
66 void set_kernel_type(KernelType kernel_type) { kernel_type_ = kernel_type; } in set_kernel_type()
|