Searched refs:wrapped_done (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 61 auto wrapped_done = [status_cb](const Status& s) { in CopyHostToDevice() local
67 edge_name, sync_dst_compute, wrapped_done = std::move(wrapped_done)]( in CopyHostToDevice()
72 dst, to, recv_dev_context, wrapped_done, in CopyHostToDevice()
88 &from, dst, to, wrapped_done, sync_dst_compute); in CopyHostToDevice()
133 auto wrapped_done = [status_cb](const Status& s) { in CopyDeviceToDevice() local
140 dev_to_dev_stream_index, wrapped_done = std::move(wrapped_done)]( in CopyDeviceToDevice()
148 dev_to_dev_stream_index, wrapped_done); in CopyDeviceToDevice()
164 dev_to_dev_stream_index, wrapped_done); in CopyDeviceToDevice()
341 auto wrapped_done = [status_cb](const Status& s) { in CopyDeviceToHost() local
347 cpu_allocator, wrapped_done = std::move(wrapped_done)]( in CopyDeviceToHost()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 490 auto wrapped_done = [&](const Status& status) { in RunComponentFunction() local
498 wrapped_done(status); in RunComponentFunction()
503 wrapped_done(AddOpRetvalsToResponse( in RunComponentFunction()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 1336 auto wrapped_done = [&](const Status& s) { in EagerKernelExecuteAsync() local
1341 wrapped_done(s); in EagerKernelExecuteAsync()
1348 wrapped_done(GetKernelOutputs(outputs.get(), num_outputs, retvals, ctx, in EagerKernelExecuteAsync()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 277 auto wrapped_done = [context, done]( in CheckLapackInfoAndDeleteSolverAsync() local
288 wrapped_done); in CheckLapackInfoAndDeleteSolverAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 670 auto wrapped_done = [to, copy, done = std::move(done)](const Status& s) { in MaybeCopyTensorToGPU() local
680 &from, this, copy, std::move(wrapped_done), in MaybeCopyTensorToGPU()
|