Searched refs: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() argument
53 output_types.emplace_back(kernel_attr.GetInputAttr(input_index).first); in UpdatePrevNotCNodeFormatDtype()
101 void GetOutputFormatsAndDtypes(const CNodePtr &kernel_node, const KernelAttr &kernel_attr, in GetOutputFormatsAndDtypes() argument
105 output_formats->emplace_back(kernel_attr.GetOutputAttr(output_index).second); in GetOutputFormatsAndDtypes()
106 auto dtype = kernel_attr.GetOutputAttr(output_index).first; in GetOutputFormatsAndDtypes()
125 int GetOutputDtypeMatchedNum(const KernelAttr &kernel_attr, const std::vector<TypeId> &output_types… in GetOutputDtypeMatchedNum() argument
126 if (kernel_attr.GetOutputSize() != output_types.size()) { in GetOutputDtypeMatchedNum()
127 MS_LOG(DEBUG) << "required output num:" << kernel_attr.GetInputSize() in GetOutputDtypeMatchedNum()
134 if (kernel_attr.GetOutputAttr(i).first != output_types[i]) { in GetOutputDtypeMatchedNum()
135 MS_LOG(DEBUG) << "required dtype:" << kernel_attr.GetOutputAttr(i).first in GetOutputDtypeMatchedNum()
[all …]
|
| /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() argument
39 (void)name_to_attr_creator_[kernel_name].emplace_back(kernel_attr, kernel_creator); in Register()
71 KernelAttr kernel_attr; in SetKernelAttrs() local
72 (void)kernel_attr.AddInputAttr(kernel::DtypeToTypeId(first_input_dtypes[i]), input_formats[i]); in SetKernelAttrs()
76 (void)kernel_attr.AddInputAttr(kernel::DtypeToTypeId(input_dtypes[i]), input_formats[i]); in SetKernelAttrs()
81 (void)kernel_attr.AddOutputAttr(kernel::DtypeToTypeId(output_dtypes[i]), output_formats[i]); in SetKernelAttrs()
84 (void)kernel_attr.SetAllSameAttr(true); in SetKernelAttrs()
86 (void)kernel_attrs->emplace_back(kernel_attr); in SetKernelAttrs()
144 bool CPUKernelFactory::CPUKernelSingleAttrCheck(const KernelAttr &kernel_attr, in CPUKernelSingleAttrCheck() argument
147 …auto dtype = kernel_attr.GetAllSame() ? kernel_attr.GetInputAttr(0).first : kernel_attr.GetInputAt… in CPUKernelSingleAttrCheck()
[all …]
|
| D | cpu_kernel_factory.h | 40 …void Register(const std::string &kernel_name, const KernelAttr &kernel_attr, CPUKernelCreator &&ke…
51 …bool CPUKernelSingleAttrCheck(const KernelAttr &kernel_attr, const KernelBuildInfo &kernel_info) c…
57 …CPUKernelRegistrar(const std::string &kernel_name, const KernelAttr &kernel_attr, CPUKernelCreator… in CPUKernelRegistrar() argument
58 CPUKernelFactory::GetInstance().Register(kernel_name, kernel_attr, std::move(kernel_creator)); in CPUKernelRegistrar()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 39 …void Register(const std::string &kernel_name, const KernelAttr &kernel_attr, GpuKernelCreater &&cr…
69 …GpuKernelRegister(const std::string &kernel_name, const KernelAttr &kernel_attr, GpuKernelCreater … in GpuKernelRegister() argument
70 GpuKernelFactory::GetInstance().Register(kernel_name, kernel_attr, std::move(creator)); in GpuKernelRegister()
|
| D | gpu_kernel_factory.cc | 34 void GpuKernelFactory::Register(const std::string &kernel_name, const KernelAttr &kernel_attr, in Register() argument
36 map_kernel_name_to_creater_[kernel_name].emplace_back(kernel_attr, creator); in Register()
|