Searched refs:RegCustomKernel (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/registry/ |
| D | register_kernel.cc | 25 Status RegisterKernel::RegCustomKernel(const std::string &arch, const std::string &provider, DataTy… in RegCustomKernel() function in mindspore::registry::RegisterKernel
28 …return RegistryKernelImpl::GetInstance()->RegCustomKernel(arch, provider, data_type, type, creator… in RegCustomKernel()
|
| D | register_kernel_impl.h | 39 Status RegCustomKernel(const std::string &arch, const std::string &provider, DataType data_type,
|
| D | register_kernel_impl.cc | 52 Status RegistryKernelImpl::RegCustomKernel(const std::string &arch, const std::string &provider, Da… in RegCustomKernel() function in mindspore::registry::RegistryKernelImpl
|
| /third_party/mindspore/mindspore/lite/include/registry/ |
| D | register_kernel.h | 77 …static Status RegCustomKernel(const std::string &arch, const std::string &provider, DataType data_…
116 RegisterKernel::RegCustomKernel(arch, provider, data_type, op_type, creator); in KernelReg()
|