Searched refs:CopyDeviceToHost (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 267 CopyDeviceToHost(input, cpu_allocator, out_allocator, edge_name, src, in ViaDMA()
276 CopyDeviceToHost(input, cpu_allocator, out_allocator, edge_name, src, in ViaDMA()
332 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator, in CopyDeviceToHost() function
351 CopyDeviceToHost(&from, cpu_allocator, out_allocator, edge_name, in CopyDeviceToHost()
|
| D | copy_tensor.h | 72 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 547 CopyDeviceToHost(&val, alloc, alloc, request->rendezvous_key(), in GrpcRecvTensorAsync()
|