Home
last modified time | relevance | path

Searched refs:eigen_sycl_device (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device_context.cc35 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
40 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
45 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
50 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
55 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
60 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
65 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
70 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
75 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
80 device->eigen_sycl_device()->memcpyHostToDevice( in CopyCPUTensorToDevice()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h76 const Eigen::SyclDevice* eigen_sycl_device() const override { in eigen_sycl_device() function
77 return underlying_->eigen_sycl_device(); in eigen_sycl_device()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h209 virtual const Eigen::SyclDevice* eigen_sycl_device() const { in eigen_sycl_device() function
Dop_kernel.h1118 const Eigen::SyclDevice& eigen_sycl_device() const { in eigen_sycl_device() function
1119 return *device()->eigen_sycl_device(); in eigen_sycl_device()
Dop_kernel.cc1385 return eigen_sycl_device(); in eigen_device()
/external/tensorflow/tensorflow/core/kernels/
Dpack_op.cc123 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat); in Compute()
Dconcat_op.cc165 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat); in Compute()
Ddebug_ops.h97 SYCLmemcpy(context->eigen_sycl_device(), src_tensor, copied_tensor); in Compute()
Dscatter_op.cc209 c->eigen_sycl_device().memcpyDeviceToHost( in DoCompute()
Dbatch_matmul_op_impl.h510 auto d = context->eigen_sycl_device();
Dscatter_nd_op.cc646 auto device = c->eigen_sycl_device(); in operator ()()
Dtraining_ops.cc532 auto device = ctx->eigen_sycl_device(); in Compute()
2941 auto device = ctx->eigen_sycl_device(); in Compute()