Searched defs:RegKernel (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/registry/ | ||
| D | register_kernel.cc | 35 Status RegisterKernel::RegKernel(const std::string &arch, const std::string &provider, DataType dat… in RegKernel() function in mindspore::registry::RegisterKernel |
| D | register_kernel_impl.cc | 93 Status RegistryKernelImpl::RegKernel(const std::string &arch, const std::string &provider, DataType… in RegKernel() function in mindspore::registry::RegistryKernelImpl |
| /third_party/mindspore/mindspore/lite/src/ | ||
| D | kernel_registry.cc | 101 void KernelRegistry::RegKernel(const KernelKey desc, const kernel::KernelCreator creator) { in RegKernel() function in mindspore::lite::KernelRegistry |