Searched refs:typed_registry (Results 1 – 1 of 1) sorted by relevance
1317 auto typed_registry = GlobalKernelRegistryTyped(); in FindKernelRegistration() local1318 tf_shared_lock lock(typed_registry->mu); in FindKernelRegistration()1319 auto regs = typed_registry->registry.equal_range(key); in FindKernelRegistration()1350 auto regs = typed_registry->registry.equal_range(default_key); in FindKernelRegistration()1537 KernelRegistry* const typed_registry = GlobalKernelRegistryTyped(); in GetFilteredRegisteredKernels() local1539 tf_shared_lock lock(typed_registry->mu); in GetFilteredRegisteredKernels()1540 kernel_list.mutable_kernel()->Reserve(typed_registry->registry.size()); in GetFilteredRegisteredKernels()1541 for (const auto& p : typed_registry->registry) { in GetFilteredRegisteredKernels()1685 auto typed_registry = GlobalKernelRegistryTyped(); in ValidateKernelRegistrations() local1686 tf_shared_lock lock(typed_registry->mu); in ValidateKernelRegistrations()[all …]