Searched refs:CopyHostToDevice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 50 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, in CopyHostToDevice() function
70 CopyHostToDevice(&from, cpu_allocator, out_allocator, edge_name, in CopyHostToDevice()
314 CopyHostToDevice(cpu_tensor, cpu_allocator, out_allocator, edge_name, in ViaDMA()
336 CopyHostToDevice(input, cpu_allocator, out_allocator, edge_name, dst, in ViaDMA()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 83 inline bool CopyHostToDevice(OpKernelContext* context, void* dst, in CopyHostToDevice() function
711 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrfBatchedImpl()
749 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
753 if (!CopyHostToDevice(context, dev_b_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
794 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetriBatchedImpl()
796 !CopyHostToDevice(context, dev_a_inv_dev_ptrs.mutable_data(), in GetriBatchedImpl()
837 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in MatInvBatchedImpl()
839 !CopyHostToDevice(context, dev_a_inv_dev_ptrs.mutable_data(), in MatInvBatchedImpl()
|