Searched defs:kernel_attr (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | kernel_select_cpu.cc | 45 void UpdatePrevNotCNodeFormatDtype(const KernelAttr &kernel_attr, const std::vector<size_t> &input_… in UpdatePrevNotCNodeFormatDtype()
101 void GetOutputFormatsAndDtypes(const CNodePtr &kernel_node, const KernelAttr &kernel_attr, in GetOutputFormatsAndDtypes()
125 int GetOutputDtypeMatchedNum(const KernelAttr &kernel_attr, const std::vector<TypeId> &output_types… in GetOutputDtypeMatchedNum()
144 int GetInputDtypeFormatMatchedNum(const KernelAttr &kernel_attr, const std::vector<TypeId> &input_t… in GetInputDtypeFormatMatchedNum()
163 void ExpandKernelAttr(const CNodePtr &kernel_node, KernelAttr *kernel_attr) { in ExpandKernelAttr()
277 for (auto kernel_attr : kernel_attrs) { in SelectKernel() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cpu_kernel_factory.cc | 37 void CPUKernelFactory::Register(const std::string &kernel_name, const KernelAttr &kernel_attr, in Register()
71 KernelAttr kernel_attr; in SetKernelAttrs() local
144 bool CPUKernelFactory::CPUKernelSingleAttrCheck(const KernelAttr &kernel_attr, in CPUKernelSingleAttrCheck()
|
| D | cpu_kernel_factory.h | 57 …CPUKernelRegistrar(const std::string &kernel_name, const KernelAttr &kernel_attr, CPUKernelCreator… in CPUKernelRegistrar()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 69 …GpuKernelRegister(const std::string &kernel_name, const KernelAttr &kernel_attr, GpuKernelCreater … in GpuKernelRegister()
|
| D | gpu_kernel_factory.cc | 34 void GpuKernelFactory::Register(const std::string &kernel_name, const KernelAttr &kernel_attr, in Register()
|