Searched refs:typed_registry (Results 1 – 1 of 1) sorted by relevance
1291 auto typed_registry = GlobalKernelRegistryTyped(); in FindKernelRegistration() local1292 tf_shared_lock lock(typed_registry->mu); in FindKernelRegistration()1293 auto regs = typed_registry->registry.equal_range(key); in FindKernelRegistration()1324 auto regs = typed_registry->registry.equal_range(default_key); in FindKernelRegistration()1497 KernelRegistry* const typed_registry = GlobalKernelRegistryTyped(); in GetFilteredRegisteredKernels() local1499 tf_shared_lock lock(typed_registry->mu); in GetFilteredRegisteredKernels()1500 kernel_list.mutable_kernel()->Reserve(typed_registry->registry.size()); in GetFilteredRegisteredKernels()1501 for (const auto& p : typed_registry->registry) { in GetFilteredRegisteredKernels()1635 auto typed_registry = GlobalKernelRegistryTyped(); in ValidateKernelRegistrations() local1636 tf_shared_lock lock(typed_registry->mu); in ValidateKernelRegistrations()[all …]