| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.h | 49 StatusCallback done) override;
53 StatusCallback done) override;
58 StatusCallback done) override;
62 StatusCallback done) override;
66 StatusCallback done) override;
70 StatusCallback done) override;
78 StatusCallback done) override;
82 StatusCallback done) override;
85 TensorResponse* response, StatusCallback done) override;
88 StatusCallback done) override;
[all …]
|
| D | test_utils.h | 35 StatusCallback done) override { in GetStatusAsync()
41 StatusCallback done) override { in CreateWorkerSessionAsync()
48 StatusCallback done) override { in DeleteWorkerSessionAsync()
54 StatusCallback done) override { in RegisterGraphAsync()
60 StatusCallback done) override { in DeregisterGraphAsync()
66 StatusCallback done) override { in RunGraphAsync()
72 StatusCallback done) override { in CleanupGraphAsync()
78 StatusCallback done) override { in CleanupAllAsync()
83 TensorResponse* response, StatusCallback done) override { in RecvTensorAsync()
88 StatusCallback done) override { in LoggingAsync()
[all …]
|
| D | worker_interface.h | 31 typedef std::function<void(const Status&)> StatusCallback; typedef
41 StatusCallback done) = 0;
45 CreateWorkerSessionResponse* response, StatusCallback done) = 0;
49 DeleteWorkerSessionResponse* response, StatusCallback done) = 0;
53 StatusCallback done) = 0;
57 StatusCallback done) = 0;
61 StatusCallback done) = 0;
64 RunGraphResponse* response, StatusCallback done) { in RunGraphAsync()
98 StatusCallback done) = 0;
102 StatusCallback done) = 0;
[all …]
|
| D | worker.cc | 34 GetStatusResponse* response, StatusCallback done) { in GetStatusAsync()
47 StatusCallback done) { in CreateWorkerSessionAsync()
57 StatusCallback done) { in DeleteWorkerSessionAsync()
64 StatusCallback done) { in RegisterGraphAsync()
85 StatusCallback done) { in DeregisterGraphAsync()
132 StatusCallback done) { in RunGraphAsync()
156 StatusCallback done) { in DoRunGraph()
265 StatusCallback done) { in DoPartialRunGraph()
348 StatusCallback done) { in CleanupGraphAsync()
365 StatusCallback done) { in CleanupAllAsync()
[all …]
|
| D | collective_param_resolver_distributed.h | 36 const StatusCallback& done) override;
41 const StatusCallback& done) override;
46 const StatusCallback& done) override;
73 const StatusCallback& done)
82 const StatusCallback& done)
|
| D | device_resolver_distributed.h | 39 const StatusCallback& done) override;
43 const StatusCallback& done) override;
50 const StatusCallback& done) LOCKS_EXCLUDED(mu_);
57 const StatusCallback& done);
|
| D | graph_mgr.h | 88 typedef std::function<void(const Status&)> StatusCallback; typedef
94 const NamedTensors& in, StatusCallback done);
99 StatusCallback done);
171 StatusCallback done);
|
| D | device_resolver_distributed.cc | 29 const StatusCallback& done) { in GetLocalityAsync()
69 std::vector<DeviceLocality>* localities, const StatusCallback& done) { in GetDeviceLocalitiesAsync()
76 std::vector<DeviceLocality>* localities, const StatusCallback& done) { in GetDeviceLocalitiesRecursive()
97 const string& device, const string& task, const StatusCallback& done) { in RefreshRemoteAttributes()
|
| D | collective_param_resolver_distributed.cc | 38 void IssueCall(const StatusCallback& done) override { in IssueCall()
71 void IssueCall(const StatusCallback& done) override { in IssueCall()
97 const StatusCallback& done) { in CompleteParamsAsync()
114 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
146 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
162 StatusCallback done_and_cleanup = [this, cp, device, done](const Status& s) { in CompleteInstanceAsync()
292 const CompleteInstanceResponse& resp, const StatusCallback& done) { in UpdateInstanceCache()
364 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceDistributed()
|
| D | partial_run_mgr.h | 66 void PartialRunDone(int step_id, StatusCallback done, const Status& status);
75 StatusCallback final_callback = nullptr;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_remote_worker.cc | 75 StatusCallback done) override { in GetStatusAsync()
81 StatusCallback done) override { in CreateWorkerSessionAsync()
88 StatusCallback done) override { in DeleteWorkerSessionAsync()
95 StatusCallback done) override { in RegisterGraphAsync()
101 StatusCallback done) override { in DeregisterGraphAsync()
106 RunGraphResponse* response, StatusCallback done) override { in RunGraphAsync()
111 StatusCallback done) override { in RunGraphAsync()
118 StatusCallback done) override { in CleanupGraphAsync()
124 StatusCallback done) override { in CleanupAllAsync()
129 RecvBufResponse* response, StatusCallback done) override { in RecvBufAsync()
[all …]
|
| D | grpc_worker_service.h | 46 StatusCallback done);
49 LoggingResponse* response, StatusCallback done);
52 RecvBufResponse* response, StatusCallback done);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 138 const StatusCallback& done) = 0;
144 const StatusCallback& done) = 0;
161 const StatusCallback& done) = 0;
168 const StatusCallback& done) = 0;
175 const StatusCallback& done) = 0;
190 const StatusCallback& done) = 0;
195 const StatusCallback& done) = 0;
242 const StatusCallback& done) = 0;
250 const StatusCallback& done) = 0;
263 const string& exec_key, StatusCallback done) { in ExecuteAsync()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 41 StatusCallback done);
50 StatusCallback done);
91 StatusCallback done);
97 int dev_to_dev_stream_index, StatusCallback done);
107 StatusCallback done);
|
| D | gpu_util_platform_specific.cc | 29 StatusCallback done) const { in CopyCPUTensorToDevice()
36 StatusCallback done) { in CopyDeviceTensorToCPU()
43 StatusCallback done) const { in CopyTensorInSameDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.h | 45 const StatusCallback& done) override;
53 const StatusCallback& done) override;
57 const StatusCallback& done) override { in GetDeviceLocalitiesAsync()
63 const StatusCallback& done) override { in GetLocalityAsync()
82 const StatusCallback& done);
|
| D | hierarchical_tree_broadcaster.h | 53 void Run(StatusCallback done) override;
71 const Tensor* src_tensor, const StatusCallback& done);
77 const StatusCallback& done);
84 StatusCallback done_;
|
| D | collective_param_resolver_local.h | 49 const StatusCallback& done) override;
54 const StatusCallback& done) override;
59 const StatusCallback& done) override;
74 std::vector<StatusCallback> waiting GUARDED_BY(mu);
173 InstanceRec* ir, const StatusCallback& done)
192 const StatusCallback& done)
201 const StatusCallback& done)
|
| D | executor.h | 150 typedef std::function<void(const Status&)> StatusCallback; typedef
160 ExecutorBarrier(size_t num, Rendezvous* r, StatusCallback done) in ExecutorBarrier()
167 StatusCallback Get() { in Get()
173 StatusCallback done_cb_ = nullptr;
181 StatusCallback done = nullptr; in WhenDone()
|
| D | test_collective_executor_mgr.h | 37 const StatusCallback& done) override { in RecvFromPeer()
47 const StatusCallback& done) override { in PostToPeer()
97 const StatusCallback& done) override { in GetStepSequenceAsync()
102 const StatusCallback& done) override { in RefreshStepIdSequenceAsync()
|
| D | copy_tensor.cc | 53 DeviceContext* recv_dev_context, StatusCallback done) { in CopyHostToDevice()
65 edge_name](StatusCallback wrapped_done_, in CopyHostToDevice()
118 DeviceContext* send_dev_context, StatusCallback done) { in CopyDeviceToHost()
130 cpu_allocator](StatusCallback wrapped_done_, in CopyDeviceToHost()
187 int dev_to_dev_stream_index, StatusCallback done) { in CopyDeviceToDevice()
200 dev_to_dev_stream_index](StatusCallback wrapped_done_, in CopyDeviceToDevice()
254 int dev_to_dev_stream_index, StatusCallback done) { in ViaDMA()
298 [cpu_tensor](StatusCallback done_, in ViaDMA()
307 out_allocator, edge_name, dst, output](StatusCallback delete_and_done_, in ViaDMA()
|
| D | base_collective_executor.h | 111 const string& exec_key, StatusCallback done) override;
115 StatusCallback done) override;
126 const StatusCallback& done) override { in RecvFromPeer()
138 const StatusCallback& done) override { in PostToPeer()
|
| D | gpu_device_context.h | 54 StatusCallback done) const override;
58 StatusCallback done) override;
62 StatusCallback done) const override;
|
| D | ring_alg.h | 88 void DispatchSend(RingField* rf, const StatusCallback& done);
89 void DispatchRecv(RingField* rf, const StatusCallback& done);
112 StatusCallback done_;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 65 StatusCallback done) const override;
68 Tensor* cpu_tensor, StatusCallback done) override;
71 StatusCallback done) const override;
|