Searched defs:pool_fn (Results 1 – 5 of 5) sorted by relevance
57 const DescriptorPoolRegistry::DescriptorPoolFn& pool_fn) { in DescriptorPoolRegistration()64 #define REGISTER_DESCRIPTOR_POOL(source, pool_fn) \ argument67 #define REGISTER_DESCRIPTOR_POOL_UNIQ_HELPER(ctr, source, pool_fn) \ argument70 #define REGISTER_DESCRIPTOR_POOL_UNIQ(ctr, source, pool_fn) \ argument
38 const DescriptorPoolRegistry::DescriptorPoolFn& pool_fn) { in Register()
69 auto pool_fn = DescriptorPoolRegistry::Global()->Get(descriptor_source); in GetDescriptorPool() local
598 def pool_fn(seqs): function715 def pool_fn(seqs): function
765 pool_fn = lambda x: random_tensor_pool.tensor_pool(x, pool_size=pool_size) function