Searched defs:RegKernel (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/registry/ |
| D | register_kernel.cc | 31 Status RegisterKernel::RegKernel(const std::vector<char> &arch, const std::vector<char> &provider, … in RegKernel() function in mindspore::registry::RegisterKernel
|
| D | register_kernel_impl.cc | 92 Status RegistryKernelImpl::RegKernel(const std::string &arch, const std::string &provider, DataType… in RegKernel() function in mindspore::registry::RegistryKernelImpl
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | kernel_registry.cc | 110 void KernelRegistry::RegKernel(const KernelKey desc, const kernel::KernelCreator creator) { in RegKernel() function in mindspore::lite::KernelRegistry
130 void KernelRegistry::RegKernel(KERNEL_ARCH arch, TypeId data_type, int op_type, kernel::KernelCreat… in RegKernel() function in mindspore::lite::KernelRegistry
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/include/registry/ |
| D | register_kernel.h | 135 Status RegisterKernel::RegKernel(const std::string &arch, const std::string &provider, DataType dat… in RegKernel() function
|