Home
last modified time | relevance | path

Searched refs:RefCountPtr (Results 1 – 25 of 52) sorted by relevance

123

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_test.cc89 core::RefCountPtr<EagerContext> ctx(new EagerContext( in TEST()
95 core::RefCountPtr<TestCustomDeviceTensorHandle> tensor( in TEST()
114 core::RefCountPtr<EagerContext> ctx(new EagerContext( in TEST()
121 core::RefCountPtr<TestCustomDeviceTensorHandle> custom_float_tensor( in TEST()
123 core::RefCountPtr<TestCustomDeviceTensorHandle> custom_resource_tensor( in TEST()
128 core::RefCountPtr<TensorHandle> physical_resource_tensor( in TEST()
133 core::RefCountPtr<TensorHandle> physical_float_tensor( in TEST()
Deager_executor.h196 void NodeDone(const core::RefCountPtr<NodeItem>& item, const Status& status,
206 Status RunItem(core::RefCountPtr<NodeItem> item, bool from_queue);
207 Status MoveToUnfinished(core::RefCountPtr<NodeItem> item, bool from_queue);
224 std::queue<core::RefCountPtr<NodeItem>> node_queue_
228 std::map<uint64, core::RefCountPtr<NodeItem>, std::less<uint64>>
Deager_executor.cc129 core::RefCountPtr<NodeItem> item(new NodeItem); in AddOrExecute()
214 void EagerExecutor::NodeDone(const core::RefCountPtr<NodeItem>& item, in NodeDone()
229 std::forward_list<core::RefCountPtr<NodeItem>> items_to_destroy; in NodeDone()
328 core::RefCountPtr<NodeItem> curr_item; in Run()
352 Status EagerExecutor::RunItem(core::RefCountPtr<NodeItem> item, in RunItem()
397 core::RefCountPtr<NodeItem> async_item(async_ref); in RunItem()
405 Status EagerExecutor::MoveToUnfinished(core::RefCountPtr<NodeItem> item, in MoveToUnfinished()
Dexecute_node.h56 const core::RefCountPtr<KernelAndDevice>& kernel);
97 const core::RefCountPtr<KernelAndDevice>& kernel, in ExecuteNode()
141 const core::RefCountPtr<KernelAndDevice>& kernel_;
153 core::RefCountPtr<KernelAndDevice> kernel, in AsyncExecuteNode()
236 core::RefCountPtr<KernelAndDevice> kernel_;
Dcontext.h245 core::RefCountPtr<KernelAndDevice> GetCachedKernel(Fprint128 cache_key);
341 core::RefCountPtr<eager::EagerClient>* client);
343 core::RefCountPtr<eager::EagerClient>* client);
345 core::RefCountPtr<eager::EagerClient>* client);
624 std::unordered_map<Fprint128, core::RefCountPtr<KernelAndDevice>,
Dcontext.cc457 core::RefCountPtr<eager::EagerClient> client; in CloseRemoteContexts()
638 core::RefCountPtr<eager::EagerClient> eager_client; in MaybeRegisterFunctionRemotely()
679 core::RefCountPtr<eager::EagerClient> eager_client; in RegisterExistingFunctionsOnRemoteWorkers()
820 core::RefCountPtr<eager::EagerClient> eager_client; in SyncExecutors()
840 core::RefCountPtr<KernelAndDevice> EagerContext::GetCachedKernel( in GetCachedKernel()
847 core::RefCountPtr<KernelAndDevice> new_ref(iter->second.get()); in GetCachedKernel()
855 core::RefCountPtr<KernelAndDevice> new_ref(kernel); in AddKernelToCache()
993 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()
998 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()
1031 core::RefCountPtr<eager::EagerClient>* client) { in GetClient()
[all …]
Dexecute.h53 const core::RefCountPtr<KernelAndDevice>& kernel,
/external/tensorflow/tensorflow/core/kernels/
Dsummary_kernels.cc49 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
78 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
104 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
129 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
153 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
189 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
209 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
231 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
259 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
294 core::RefCountPtr<SummaryWriterInterface> s; in Compute()
[all …]
Dresource_variable_ops.cc134 core::RefCountPtr<Var> variable; in Compute()
173 std::vector<core::RefCountPtr<Var>> variables(dtypes_.size()); in Compute()
359 core::RefCountPtr<Var> variable; in Compute()
429 core::RefCountPtr<Var> variable; in Compute()
525 core::RefCountPtr<Var> variable; in Compute()
592 core::RefCountPtr<Var> variable; in Compute()
622 core::RefCountPtr<Var> v; in Compute()
788 core::RefCountPtr<Var> v; in Compute()
850 core::RefCountPtr<Var> v; in Compute()
870 core::RefCountPtr<Var> v; in DoCompute()
/external/tensorflow/tensorflow/c/experimental/op_handler/
Dinternal_test.cc38 core::RefCountPtr<OpHandler>& merged_handler) override { in Merge()
43 core::RefCountPtr<OpHandler> next_handler_ = nullptr;
64 core::RefCountPtr<TestOpHandler> outer_handler(new TestOpHandler()); in TEST()
65 core::RefCountPtr<TestOpHandler> inner_handler(new TestOpHandler()); in TEST()
82 core::RefCountPtr<OpHandler> merged; in TEST()
Dinternal.h52 core::RefCountPtr<OpHandler>& merged_handler) = 0;
75 core::RefCountPtr<OpHandler> handler_;
104 core::RefCountPtr<OpHandler> default_handler_;
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dresource_ops.cc59 core::RefCountPtr<TensorForestTreeResource> decision_tree_resource; in Compute()
77 core::RefCountPtr<TensorForestTreeResource> decision_tree_resource; in Compute()
103 core::RefCountPtr<TensorForestTreeResource> decision_tree_resource; in Compute()
Dprediction_ops.cc33 core::RefCountPtr<TensorForestTreeResource> decision_tree_resource; in Compute()
78 const core::RefCountPtr<TensorForestTreeResource>& decision_tree_resource, in set_output_value()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Ddestroy_tensor_handle_node.h33 core::RefCountPtr<EagerClient> eager_client, in DestroyTensorHandleNode()
80 core::RefCountPtr<EagerClient> eager_client_;
Dremote_copy_node.cc44 EagerOperation* op, core::RefCountPtr<KernelAndDevice>* kernel) { in CreateUncachedKernelAndDeviceOp()
104 core::RefCountPtr<KernelAndDevice> kernel; in RunLocalSend()
164 core::RefCountPtr<eager::EagerClient> eager_client; in StartSend()
192 core::RefCountPtr<KernelAndDevice> kernel; in RunLocalRecv()
222 core::RefCountPtr<eager::EagerClient> eager_client; in RunRemoteRecv()
385 core::RefCountPtr<eager::EagerClient> eager_client; in StartSendPackedHandle()
439 core::RefCountPtr<eager::EagerClient> eager_client; in StartRemoteSendTensor()
Dcluster_function_library_runtime.h89 core::RefCountPtr<EagerClient> eager_client;
97 eager_client(core::RefCountPtr<EagerClient>(eager_client)), in FunctionData()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_client.cc248 core::RefCountPtr<EagerClient>* client) override { in GetClient()
260 core::RefCountPtr<EagerClient> worker( in GetClient()
292 std::unordered_map<string, core::RefCountPtr<EagerClient>> clients_
294 std::vector<core::RefCountPtr<GrpcEagerClientThread>> threads_;
Dgrpc_eager_client_test.cc45 core::RefCountPtr<EagerClient> eager_client; in TEST()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresource_ops.cc82 core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource; in Compute()
144 core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource; in Compute()
171 core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource; in Compute()
Dtraining_ops.cc47 core::RefCountPtr<BoostedTreesEnsembleResource> ensemble_resource; in Compute()
150 const core::RefCountPtr<BoostedTreesEnsembleResource>& resource) { in UpdateGlobalAttemptsAndRetrieveGrowableTree()
243 core::RefCountPtr<BoostedTreesEnsembleResource> ensemble_resource; in Compute()
370 const core::RefCountPtr<BoostedTreesEnsembleResource>& resource) { in UpdateGlobalAttemptsAndRetrieveGrowableTree()
464 core::RefCountPtr<BoostedTreesEnsembleResource> ensemble_resource; in Compute()
Dquantile_ops.cc230 core::RefCountPtr<QuantileStreamResource> stream_resource; in Compute()
288 core::RefCountPtr<QuantileStreamResource> stream_resource; in Compute()
350 core::RefCountPtr<QuantileStreamResource> streams_resource; in Compute()
404 core::RefCountPtr<QuantileStreamResource> stream_resource; in Compute()
459 core::RefCountPtr<QuantileStreamResource> stream_resource; in Compute()
Dprediction_ops.cc68 core::RefCountPtr<BoostedTreesEnsembleResource> resource; in Compute()
211 core::RefCountPtr<BoostedTreesEnsembleResource> resource; in Compute()
311 core::RefCountPtr<BoostedTreesEnsembleResource> resource; in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dstats_aggregator_ops.cc263 core::RefCountPtr<StatsAggregatorResource> resource; in Compute()
285 core::RefCountPtr<StatsAggregatorResource> resource; in Compute()
293 core::RefCountPtr<SummaryWriterInterface> summary_resource; in Compute()
/external/tensorflow/tensorflow/core/framework/
Dresource_mgr_test.cc276 core::RefCountPtr<StubResource> r; in TEST()
283 core::RefCountPtr<StubResource> unused; in TEST()
347 core::RefCountPtr<StubResource> lookup_r; in TEST()
Dresource_mgr.h240 core::RefCountPtr<ResourceBase> resource;
358 core::RefCountPtr<T>* value);
364 std::vector<core::RefCountPtr<T>>* values);
382 core::RefCountPtr<T>* value,
752 core::RefCountPtr<T>* value) {
763 std::vector<core::RefCountPtr<T>>* values) {
783 core::RefCountPtr<T>* value,

123