| /external/tensorflow/tensorflow/core/kernels/ |
| D | roll_op_gpu.cu.cc | 73 d.memcpyHostToDevice(dim_buf, dim_size.data(), dim_bytes); in operator ()()
74 d.memcpyHostToDevice(thres_buf, threshold.data(), thres_bytes); in operator ()()
75 d.memcpyHostToDevice(range_buf, dim_range.data(), range_bytes); in operator ()()
|
| D | tile_functor_gpu.h | 73 d.memcpyHostToDevice(dev_buf, host_buf.data(), num_bytes); in TileSimple()
|
| D | transpose_functor_gpu.cu.cc | 79 d.memcpyHostToDevice(dev_buf, host_buf.data(), num_bytes); in TransposeSimple()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_sycl.cpp | 45 …sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float)); in test_full_reductions_sycl()
80 …sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float)); in test_first_dim_reductions_sycl()
118 …sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float)); in test_last_dim_reductions_sycl()
|
| D | cxx11_tensor_forced_eval_sycl.cpp | 46 …sycl_device.memcpyHostToDevice(gpu_in1_data, in1.data(),(in1.dimensions().TotalSize())*sizeof(floa… in test_forced_eval_sycl()
47 …sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in1.dimensions().TotalSize())*sizeof(floa… in test_forced_eval_sycl()
|
| D | cxx11_tensor_reduction_cuda.cu | 41 gpu_device.memcpyHostToDevice(gpu_in_ptr, in.data(), in_bytes); in test_full_reductions()
83 dev.memcpyHostToDevice(in_data, in.data(), in.size()*sizeof(Type)); in test_first_dim_reductions()
123 dev.memcpyHostToDevice(in_data, in.data(), in.size()*sizeof(Type)); in test_last_dim_reductions()
|
| D | cxx11_tensor_complex_cuda.cu | 92 gpu_device.memcpyHostToDevice(gpu_in_ptr, in.data(), in_bytes); in test_cuda_sum_reductions()
129 gpu_device.memcpyHostToDevice(gpu_in_ptr, in.data(), in_bytes); in test_cuda_product_reductions()
|
| D | cxx11_tensor_sycl.cpp | 80 …sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in2.dimensions().TotalSize())*sizeof(floa… in test_sycl_cpu()
137 …sycl_device.memcpyHostToDevice(gpu_in3_data, in3.data(),(in3.dimensions().TotalSize())*sizeof(floa… in test_sycl_cpu()
|
| D | cxx11_tensor_broadcast_sycl.cpp | 52 …sycl_device.memcpyHostToDevice(gpu_in_data, input.data(),(input.dimensions().TotalSize())*sizeof(f… in test_broadcast_sycl()
|
| D | cxx11_tensor_cast_float16_cuda.cu | 43 gpu_device.memcpyHostToDevice(d_float, floats.data(), num_elem*sizeof(float)); in test_cuda_conversion()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorDeviceDefault.h | 27 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void memcpyHostToDevice(void* dst, const void* src, size_t n… in memcpyHostToDevice() function
|
| D | TensorDeviceSycl.h | 97 …template<typename T> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void memcpyHostToDevice(T *dst, const T… in memcpyHostToDevice() function
|
| D | TensorDeviceThreadPool.h | 122 EIGEN_STRONG_INLINE void memcpyHostToDevice(void* dst, const void* src, size_t n) const { in memcpyHostToDevice() function
|
| D | TensorDeviceCuda.h | 224 EIGEN_STRONG_INLINE void memcpyHostToDevice(void* dst, const void* src, size_t n) const { in initializeDeviceProp() function
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 728 device.memcpyHostToDevice((num_outputs_t->flat<int>().data()), &num_outputs, in Compute()
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 998 cc_ctx->eigen_gpu_device().memcpyHostToDevice(data, values, in set_tensor_data()
|