Searched refs:host_copy (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 434 Tensor* host_copy = new Tensor(alloc, tensor.dtype(), tensor.shape()); in TransportOptionsFromTensor() local
436 device, device_context, &tensor, host_copy, in TransportOptionsFromTensor()
437 [done, host_copy, mutable_transport_options, this](const Status& s) { in TransportOptionsFromTensor()
440 delete host_copy; in TransportOptionsFromTensor()
443 auto buffer = DMAHelper::buffer(host_copy); in TransportOptionsFromTensor()
450 delete host_copy; in TransportOptionsFromTensor()
463 checksum = GPUUtil::Checksum(*host_copy); in TransportOptionsFromTensor()
476 delete host_copy; in TransportOptionsFromTensor()
532 Tensor host_copy; in TensorFromTransportOptions() local
536 host_copy = Tensor(alloc, tensor->dtype(), tensor->shape()); in TensorFromTransportOptions()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 194 auto host_copy = dev_lapack_info.CopyToHost(&success); in CheckLapackInfoAndDeleteSolverAsync() local
200 host_lapack_infos.push_back(std::move(host_copy)); in CheckLapackInfoAndDeleteSolverAsync()
|