Searched refs:CopyHostToDevice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 104 inline bool CopyHostToDevice(OpKernelContext* context, void* dst, in CopyHostToDevice() function
423 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in PotrfBatchedImpl()
769 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrfBatchedImpl()
807 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
811 if (!CopyHostToDevice(context, dev_b_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
852 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetriBatchedImpl()
854 !CopyHostToDevice(context, dev_a_inv_dev_ptrs.mutable_data(), in GetriBatchedImpl()
895 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in MatInvBatchedImpl()
897 !CopyHostToDevice(context, dev_a_inv_dev_ptrs.mutable_data(), in MatInvBatchedImpl()
992 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in TrsmBatchedImpl()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 51 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, in CopyHostToDevice() function
71 CopyHostToDevice(&from, cpu_allocator, out_allocator, edge_name, in CopyHostToDevice()
263 CopyHostToDevice(cpu_tensor, cpu_allocator, out_allocator, edge_name, in ViaDMA()
284 CopyHostToDevice(input, cpu_allocator, out_allocator, edge_name, dst, in ViaDMA()
|