| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.h | 49 StatusCallback done) override;
53 StatusCallback done) override;
57 StatusCallback done) override;
61 StatusCallback done) override;
65 StatusCallback done) override;
69 StatusCallback done) override;
77 StatusCallback done) override;
81 StatusCallback done) override;
84 TensorResponse* response, StatusCallback done) override;
87 StatusCallback done) override;
[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 | 32 GetStatusResponse* response, StatusCallback done) { in GetStatusAsync()
45 StatusCallback done) { in CreateWorkerSessionAsync()
54 StatusCallback done) { in DeleteWorkerSessionAsync()
61 StatusCallback done) { in RegisterGraphAsync()
73 StatusCallback done) { in DeregisterGraphAsync()
111 StatusCallback done) { in RunGraphAsync()
135 StatusCallback done) { in DoRunGraph()
208 StatusCallback done) { in DoPartialRunGraph()
286 StatusCallback done) { in CleanupGraphAsync()
294 StatusCallback done) { in CleanupAllAsync()
[all …]
|
| D | graph_mgr.h | 87 typedef std::function<void(const Status&)> StatusCallback; typedef
93 const NamedTensors& in, StatusCallback done);
98 StatusCallback done);
167 StatusCallback done);
|
| D | partial_run_mgr.h | 66 void PartialRunDone(int step_id, StatusCallback done, const Status& status);
75 StatusCallback final_callback = nullptr;
|
| D | worker_cache.h | 27 typedef std::function<void(const Status&)> StatusCallback; typedef
71 StatusCallback done) = 0;
|
| D | partial_run_mgr.cc | 46 StatusCallback done; in ExecutorDone()
72 void PartialRunMgr::PartialRunDone(int step_id, StatusCallback done, in PartialRunDone()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_remote_worker.cc | 65 StatusCallback done) override { in GetStatusAsync()
71 StatusCallback done) override { in CreateWorkerSessionAsync()
77 StatusCallback done) override { in DeleteWorkerSessionAsync()
83 StatusCallback done) override { in RegisterGraphAsync()
89 StatusCallback done) override { in DeregisterGraphAsync()
94 RunGraphResponse* response, StatusCallback done) override { in RunGraphAsync()
99 StatusCallback done) override { in RunGraphAsync()
106 StatusCallback done) override { in CleanupGraphAsync()
112 StatusCallback done) override { in CleanupAllAsync()
117 TensorResponse* response, StatusCallback done) override { in RecvTensorAsync()
[all …]
|
| D | grpc_state.h | 39 Response* response, StatusCallback done, CallOptions* call_opts) in RPCState()
46 Response* response, StatusCallback done, CallOptions* call_opts, in RPCState()
94 StatusCallback done_;
|
| D | grpc_worker_service.h | 41 StatusCallback done);
44 LoggingResponse* response, StatusCallback done);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 43 StatusCallback done);
52 StatusCallback done);
94 StatusCallback done);
102 StatusCallback done);
112 StatusCallback done);
|
| D | gpu_util_platform_specific.cc | 28 StatusCallback done) const { in CopyCPUTensorToDevice()
35 StatusCallback done) { in CopyDeviceTensorToCPU()
|
| D | gpu_util.cc | 120 StatusCallback done) { in SetProtoFromGPU()
196 StatusCallback done) { in DeviceToDeviceCopy()
262 StatusCallback done) { in CopyGPUTensorToCPU()
307 StatusCallback done) { in CopyCPUTensorToGPU()
430 StatusCallback done) { in CopyGPUTensorToSameGPU()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 55 Tensor* device_tensor, StatusCallback done) const;
58 Tensor* cpu_tensor, StatusCallback done);
76 StatusCallback done) const override;
79 Tensor* cpu_tensor, StatusCallback done) override;
|
| D | xla_device_context.cc | 60 StatusCallback done) const { in CopyCPUTensorToDevice()
96 StatusCallback done) { in CopyDeviceTensorToCPU()
133 StatusCallback done) const { in CopyCPUTensorToDevice()
140 StatusCallback done) { in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.h | 155 typedef std::function<void(const Status&)> StatusCallback; typedef
165 ExecutorBarrier(size_t num, Rendezvous* r, StatusCallback done) in ExecutorBarrier()
172 StatusCallback Get() { in Get()
178 StatusCallback done_cb_ = nullptr;
187 StatusCallback done = nullptr; in WhenDone()
|
| D | copy_tensor.cc | 53 DeviceContext* recv_dev_context, StatusCallback done) { in CopyHostToDevice()
65 StatusCallback wrapped_done_, in CopyHostToDevice()
111 DeviceContext* send_dev_context, StatusCallback done) { in CopyDeviceToHost()
123 StatusCallback wrapped_done_, in CopyDeviceToHost()
173 StatusCallback done) { in CopyDeviceToDevice()
186 status_cb](StatusCallback wrapped_done_, in CopyDeviceToDevice()
239 StatusCallback done) { in ViaDMA()
283 [cpu_tensor](StatusCallback done_, in ViaDMA()
292 out_allocator, edge_name, dst, output](StatusCallback delete_and_done_, in ViaDMA()
|
| D | rendezvous_util.h | 25 typedef std::function<void(const Status&)> StatusCallback; typedef
45 const StatusCallback& done);
|
| D | copy_tensor.h | 37 StatusCallback done);
49 const Tensor* input, Tensor* output, StatusCallback done);
|
| D | rendezvous_mgr.h | 79 typedef std::function<void(const Status&)> StatusCallback; typedef
83 Tensor* out, StatusCallback done);
|
| D | gpu_device_context.h | 57 StatusCallback done) const override;
61 StatusCallback done) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device_context.h | 36 StatusCallback done) const override;
40 StatusCallback done) override;
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.h | 45 StatusCallback done) = 0;
52 StatusCallback done) = 0;
|
| D | gdr_worker.cc | 51 StatusCallback done) { in GrpcRecvTensorAsync()
131 StatusCallback copy_ready = [response, done, copy, in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | reffed_status_callback.h | 31 explicit ReffedStatusCallback(StatusCallback done) in ReffedStatusCallback()
55 StatusCallback done_;
|