Searched refs:KernelRegistry (Results 1 – 13 of 13) sorted by relevance
7 KernelRegistry& getKernelRegistry() { in getKernelRegistry()8 static KernelRegistry kernel_registry; in getKernelRegistry()16 bool KernelRegistry::register_kernels( in register_kernels()28 bool KernelRegistry::hasKernelFn(const char* name) { in hasKernelFn()37 KernelFunction& KernelRegistry::getKernelFn(const char* name) { in getKernelFn()
50 struct KernelRegistry { struct52 KernelRegistry() : kernelRegSize_(0) {} in KernelRegistry() function
28 void RegisterTfdDelegateKernels(::tfrt::KernelRegistry* registry);29 void RegisterBatchFallbackKernels(tfrt::KernelRegistry* registry);
188 void RegisterBatchFallbackKernels(tfrt::KernelRegistry* reg) { in RegisterBatchFallbackKernels()
1261 void RegisterTfdDelegateKernels(tfrt::KernelRegistry* registry) { in RegisterTfdDelegateKernels()
1143 struct KernelRegistry { struct1253 void SetupOrDisableJit(KernelRegistry* registry) { in SetupOrDisableJit()1299 static KernelRegistry* global_kernel_registry = []() { in GlobalKernelRegistry()1300 KernelRegistry* registry = new KernelRegistry; in GlobalKernelRegistry()1307 static KernelRegistry* GlobalKernelRegistryTyped() { in GlobalKernelRegistryTyped()1311 auto* registry = reinterpret_cast<KernelRegistry*>(GlobalKernelRegistry()); in GlobalKernelRegistryTyped()1336 reinterpret_cast<KernelRegistry*>(GlobalKernelRegistry()); in InitInternal()1605 KernelRegistry* const typed_registry = GlobalKernelRegistryTyped(); in GetFilteredRegisteredKernels()
24 void RegisterKernelFallbackKernels(::tfrt::KernelRegistry* registry);
119 void RegisterKernelFallbackKernels(tfrt::KernelRegistry* registry) { in RegisterKernelFallbackKernels()
967 void RegisterKernelFallbackCompatKernels(tfrt::KernelRegistry* registry) { in RegisterKernelFallbackCompatKernels()
23 void RegisterTfJitRuntimeKernels(::tfrt::KernelRegistry* registry);
22 void RegisterTfCpuRuntimeKernels(::tfrt::KernelRegistry* registry);
67 using ::tfrt::KernelRegistry;447 void RegisterTfCpuRuntimeKernels(KernelRegistry* registry) { in RegisterTfCpuRuntimeKernels()
92 using ::tfrt::KernelRegistry;887 void RegisterTfJitRuntimeKernels(KernelRegistry* registry) { in RegisterTfJitRuntimeKernels()