Searched refs:pool_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/proto/ |
| D | descriptor_pool_registry.h | 45 void Register(const string& source, const DescriptorPoolFn& pool_fn);
57 const DescriptorPoolRegistry::DescriptorPoolFn& pool_fn) { in DescriptorPoolRegistration() argument
58 DescriptorPoolRegistry::Global()->Register(source, pool_fn); in DescriptorPoolRegistration()
64 #define REGISTER_DESCRIPTOR_POOL(source, pool_fn) \ argument
65 REGISTER_DESCRIPTOR_POOL_UNIQ_HELPER(__COUNTER__, source, pool_fn)
67 #define REGISTER_DESCRIPTOR_POOL_UNIQ_HELPER(ctr, source, pool_fn) \ argument
68 REGISTER_DESCRIPTOR_POOL_UNIQ(ctr, source, pool_fn)
70 #define REGISTER_DESCRIPTOR_POOL_UNIQ(ctr, source, pool_fn) \ argument
72 descriptor_pool_registration_fn_##ctr(source, pool_fn)
|
| D | descriptors.cc | 94 auto pool_fn = DescriptorPoolRegistry::Global()->Get(descriptor_source); in GetDescriptorPool() local
95 if (pool_fn != nullptr) { in GetDescriptorPool()
96 return (*pool_fn)(desc_pool, owned_desc_pool); in GetDescriptorPool()
|
| D | descriptor_pool_registry.cc | 38 const DescriptorPoolRegistry::DescriptorPoolFn& pool_fn) { in Register() argument
42 fns_.insert(std::pair<const string&, DescriptorPoolFn>(source, pool_fn)); in Register()
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | data_utils.py | 719 def pool_fn(seqs): function
726 return pool_fn
861 def pool_fn(seqs): function
867 return pool_fn
|