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

  /external/tensorflow/tensorflow/core/common_runtime/sycl/
sycl_device_context.cc 113 device->eigen_sycl_device()->memcpyDeviceToHost(
118 device->eigen_sycl_device()->memcpyDeviceToHost(
123 device->eigen_sycl_device()->memcpyDeviceToHost(
128 device->eigen_sycl_device()->memcpyDeviceToHost(
133 device->eigen_sycl_device()->memcpyDeviceToHost(
138 device->eigen_sycl_device()->memcpyDeviceToHost(
143 device->eigen_sycl_device()->memcpyDeviceToHost(
148 device->eigen_sycl_device()->memcpyDeviceToHost(
153 device->eigen_sycl_device()->memcpyDeviceToHost(
158 device->eigen_sycl_device()->memcpyDeviceToHost(
    [all...]
  /external/eigen/unsupported/test/
cxx11_tensor_sycl.cpp 56 sycl_device.memcpyDeviceToHost(in1.data(), gpu_in1_data ,(in1.dimensions().TotalSize())*sizeof(float));
68 sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data ,(out.dimensions().TotalSize())*sizeof(float));
82 sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(float));
96 sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(float));
110 sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(float));
124 sycl_device.memcpyDeviceToHost(out.data(),gpu_out_data,(out.dimensions().TotalSize())*sizeof(float));
139 sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(float));
cxx11_tensor_reduction_sycl.cpp 47 sycl_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_data, sizeof(float));
82 sycl_device.memcpyDeviceToHost(redux_gpu.data(), gpu_out_data, redux_gpu.dimensions().TotalSize()*sizeof(float));
120 sycl_device.memcpyDeviceToHost(redux_gpu.data(), gpu_out_data, redux_gpu.dimensions().TotalSize()*sizeof(float));
cxx11_tensor_broadcast_sycl.cpp 54 sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(float));
cxx11_tensor_forced_eval_sycl.cpp 50 sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(float));
  /external/tensorflow/tensorflow/core/kernels/
quantize_and_dequantize_op.h 49 d.memcpyDeviceToHost(&min_range, input_min.data(), sizeof(T));
50 d.memcpyDeviceToHost(&max_range, input_max.data(), sizeof(T));
reverse_sequence_op.cc 53 context->eigen_device<Device>().memcpyDeviceToHost(
scatter_op.cc 193 c->eigen_sycl_device().memcpyDeviceToHost(
scatter_nd_op.cc 424 device.memcpyDeviceToHost(indices_host_, static_cast<const Index*>(src_ptr),
training_ops.cc 472 device.memcpyDeviceToHost(&alpha, static_cast<const T*>(src_ptr), size);
    [all...]
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorDeviceDefault.h 30 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void memcpyDeviceToHost(void* dst, const void* src, size_t n) const {
TensorDeviceSycl.h 102 template<typename T> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void memcpyDeviceToHost(T *dst, const T *src, size_t n) const {
TensorDeviceCuda.h 231 EIGEN_STRONG_INLINE void memcpyDeviceToHost(void* dst, const void* src, size_t n) const {
TensorDeviceThreadPool.h 125 EIGEN_STRONG_INLINE void memcpyDeviceToHost(void* dst, const void* src, size_t n) const {

Completed in 174 milliseconds