| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | assign_sub_gpu_kernel.cc | 31 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
63 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Resize() local
|
| D | assign_add_gpu_kernel.cc | 32 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
64 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Resize() local
|
| D | correlate_gpu_kernel.cc | 31 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
53 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | kernel_select_cpu.cc | 138 bool OutputDtypeMatched(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> &output_ty… in OutputDtypeMatched()
158 bool InputDtypeFormatMatched(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> &inpu… in InputDtypeFormatMatched()
178 void ExpandKernelAttr(const CNodePtr &kernel_node, kernel::KernelAttr *kernel_attr) { in ExpandKernelAttr()
225 kernel::KernelAttr *kernel_attr) { in ExpandMultiDynamicAttr()
277 void ExpandKernelAttrByDynamicSize(const CNodePtr &kernel_node, kernel::KernelAttr *kernel_attr, bo… in ExpandKernelAttrByDynamicSize()
342 auto kernel_attr = kernel_attrs[attr_index]; in GetSupportedTypesStr() local
499 std::shared_ptr<mindspore::kernel::OpInfo> kernel_attr = nullptr; in UpdateCustomKernelBuildInfo() local
563 const kernel::KernelAttr &kernel_attr) { in FillNoneInKernelAttr()
631 for (auto kernel_attr : kernel_attrs) { in SelectKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | adagrad_v2_gpu_kernel.cc | 31 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in InOutputResize() local
77 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | adadelta_gpu_kernel.cc | 24 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in InOutputResize() local
81 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | momentum_gpu_kernel.h | 49 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | log_softmax_grad_gpu_kernel.cc | 26 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | celu_gpu_kernel.cc | 33 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | apply_gradient_descent_gpu_kernel.h | 52 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | sort_gpu_kernel.h | 83 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Resize() local
119 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | zeroslike_gpu_kernel.cc | 34 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | common_utils.cc | 181 std::vector<TypeId> GetOutputObjectTypeListFromKernelAttr(const kernel::KernelAttr &kernel_attr) { in GetOutputObjectTypeListFromKernelAttr()
190 std::vector<TypeId> GetInputObjectTypeListFromKernelAttr(const kernel::KernelAttr &kernel_attr) { in GetInputObjectTypeListFromKernelAttr()
480 std::ostream &operator<<(std::ostream &os, KernelAttr kernel_attr) { in operator <<()
515 std::pair<bool, size_t> MatchMultiDynamicKernelAttr(const KernelAttr &kernel_attr, in MatchMultiDynamicKernelAttr()
608 std::pair<bool, size_t> MatchKernelAttr(const KernelAttr &kernel_attr, in MatchKernelAttr()
648 std::pair<bool, size_t> MatchKernelAttrStrict(const KernelAttr &kernel_attr, in MatchKernelAttrStrict()
724 KernelAttr kernel_attr; in GetKernelAttrFromBuildInfo() local
744 KernelAttr kernel_attr; in GetKernelAttrFromTensors() local
763 auto kernel_attr = GetKernelAttrFromBuildInfo(build_info); in SetCpuRefMapToKernelInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/ |
| D | kernel_build_utils.cc | 104 int GetOutputDtypeMatchedNum(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> &outp… in GetOutputDtypeMatchedNum()
123 int GetInputDtypeFormatMatchedNum(const kernel::KernelAttr &kernel_attr, const std::vector<TypeId> … in GetInputDtypeFormatMatchedNum()
142 void ExpandKernelAttr(const CNodePtr &kernel_node, kernel::KernelAttr *kernel_attr) { in ExpandKernelAttr()
230 const kernel::KernelAttr &kernel_attr) { in FillNoneInKernelAttr()
297 for (auto kernel_attr : kernel_attrs) { in SelectKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/ |
| D | sparse_tensor_to_csr_sparse_matrix_gpu_kernel.cc | 40 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
64 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | get_tuple_index_info_cpu_kernel.cc | 37 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
128 auto kernel_attr = KernelAttr(); in GetOpSupport() local
|
| D | ones_cpu_kernel.h | 50 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in SelectKernelFunc() local
|
| D | zeros_cpu_kernel.h | 50 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in SelectKernelFunc() local
|
| D | broadcast_cpu_kernel_collective.cc | 38 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | allgather_cpu_kernel_collective.cc | 38 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | allreduce_cpu_kernel.cc | 42 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | opaque_predicate_kernel.cc | 46 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/ |
| D | add_bisheng_kernel.cc | 32 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| D | wkv_bisheng_kernel.cc | 31 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/other/ |
| D | dynamic_reshape_gpu_kernel.h | 43 auto kernel_attr = GetKernelAttrFromTensors(inputs, outputs); in Init() local
|