Searched refs:pool_fn (Results 1 – 6 of 6) 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 | 69 auto pool_fn = DescriptorPoolRegistry::Global()->Get(descriptor_source); in GetDescriptorPool() local
70 if (pool_fn != nullptr) { in GetDescriptorPool()
71 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 | 598 def pool_fn(seqs): function
602 return pool_fn
715 def pool_fn(seqs): function
719 return pool_fn
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | pruning.py | 456 pool_fn = pruning_utils.factorized_pool
459 pool_fn = nn_ops.pool
466 pooled_weights = pool_fn(
|
| /external/tensorflow/tensorflow/contrib/gan/python/ |
| D | train_test.py | 765 pool_fn = lambda x: random_tensor_pool.tensor_pool(x, pool_size=pool_size) function
766 new_model = train._tensor_pool_adjusted_model(model, pool_fn)
|