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

  /external/tensorflow/tensorflow/core/kernels/
cuda_solvers.cc 83 inline bool CopyHostToDevice(OpKernelContext* context, void* dst,
667 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */,
    [all...]
  /external/tensorflow/tensorflow/core/common_runtime/
copy_tensor.cc 50 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator,
299 CopyHostToDevice(cpu_tensor, cpu_allocator, out_allocator, edge_name,
321 CopyHostToDevice(input, cpu_allocator, out_allocator, edge_name, dst,

Completed in 226 milliseconds