HomeSort by relevance Sort by last modified time
    Searched refs:host_copy (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/gdr/
gdr_memory_manager.cc 434 Tensor* host_copy = new Tensor(alloc, tensor.dtype(), tensor.shape()); local
436 device, device_context, &tensor, host_copy,
437 [done, host_copy, mutable_transport_options, this](const Status& s) {
440 delete host_copy;
443 auto buffer = DMAHelper::buffer(host_copy);
450 delete host_copy;
463 checksum = GPUUtil::Checksum(*host_copy);
476 delete host_copy;
532 Tensor host_copy; local
536 host_copy = Tensor(alloc, tensor->dtype(), tensor->shape())
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
cuda_solvers.cc 194 auto host_copy = dev_lapack_info.CopyToHost(&success); local
200 host_lapack_infos.push_back(std::move(host_copy));
    [all...]

Completed in 192 milliseconds