Searched refs:CopyDeviceToDevice (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 78 void CopyDeviceToDevice(OpKernelContext* context, const Scalar* src, in CopyDeviceToDevice() function
163 CopyDeviceToDevice(context, rhs.data(), x.data(), m); in ComputeMatrix()
306 CopyDeviceToDevice(context, rhs_data, output->flat<Scalar>().data(), in ComputeWithGtsvBatched()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 120 void CopyDeviceToDevice(CopyTensor::CopyFunction copy_function, in CopyDeviceToDevice() function
145 CopyDeviceToDevice(copy_function, cpu_allocator, out_allocator, in CopyDeviceToDevice()
234 CopyDeviceToDevice(ri.copy_function, cpu_allocator, out_allocator, in ViaDMA()
|