Searched refs:fns_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/rpc/ |
| D | rpc_factory_registry.cc | 31 auto found = fns_.find(protocol); in Get()
32 if (found == fns_.end()) return nullptr; in Get()
41 fns_.insert(std::pair<const string&, RPCFactoryFn>(protocol, factory_fn)); in Register()
|
| D | rpc_factory_registry.h | 45 std::map<string, RPCFactoryFn> fns_;
|
| /external/tensorflow/tensorflow/core/util/proto/ |
| D | descriptor_pool_registry.cc | 31 auto found = fns_.find(source); in Get()
32 if (found == fns_.end()) return nullptr; in Get()
42 fns_.insert(std::pair<const string&, DescriptorPoolFn>(source, pool_fn)); in Register()
|
| D | descriptor_pool_registry.h | 48 std::map<string, DescriptorPoolFn> fns_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unary_ops_composition.cc | 101 support_.ExportComputeFns(op_names_, &fns_, &cost_)); in UnaryOpsComposition()
116 const std::size_t num_fns = fns_.size(); in Compute()
124 fns_[0](in_slice, &out_slice); in Compute()
126 fns_[i](scratch_slice, &out_slice); in Compute()
154 std::vector<ComputeFn> fns_; member in tensorflow::UnaryOpsComposition
|