| /external/tensorflow/tensorflow/core/util/proto/ |
| descriptor_pool_registry.h | 45 void Register(const string& source, const DescriptorPoolFn& pool_fn);
57 const DescriptorPoolRegistry::DescriptorPoolFn& pool_fn) {
58 DescriptorPoolRegistry::Global()->Register(source, pool_fn);
64 #define REGISTER_DESCRIPTOR_POOL(source, pool_fn) \
65 REGISTER_DESCRIPTOR_POOL_UNIQ_HELPER(__COUNTER__, source, pool_fn)
67 #define REGISTER_DESCRIPTOR_POOL_UNIQ_HELPER(ctr, source, pool_fn) \
68 REGISTER_DESCRIPTOR_POOL_UNIQ(ctr, source, pool_fn)
70 #define REGISTER_DESCRIPTOR_POOL_UNIQ(ctr, source, pool_fn) \
72 descriptor_pool_registration_fn_##ctr(source, pool_fn)
|
| descriptor_pool_registry.cc | 38 const DescriptorPoolRegistry::DescriptorPoolFn& pool_fn) {
42 fns_.insert(std::pair<const string&, DescriptorPoolFn>(source, pool_fn));
|
| descriptors.cc | 69 auto pool_fn = DescriptorPoolRegistry::Global()->Get(descriptor_source); local
70 if (pool_fn != nullptr) {
71 return (*pool_fn)(desc_pool, owned_desc_pool);
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| data_utils.py | 598 def pool_fn(seqs): function in function:OrderedEnqueuer._get_executor_init
602 return pool_fn
715 def pool_fn(seqs): function in function:GeneratorEnqueuer._get_executor_init
719 return pool_fn
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| 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/ |
| train_test.py | 765 pool_fn = lambda x: random_tensor_pool.tensor_pool(x, pool_size=pool_size)
766 new_model = train._tensor_pool_adjusted_model(model, pool_fn)
[all...] |
