Home
last modified time | relevance | path

Searched refs:get_sycl_buffer (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_allocator.cc46 const auto& allocated_buffer = sycl_device_->get_sycl_buffer(p); in AllocateRaw()
62 const auto& buffer_to_delete = sycl_device_->get_sycl_buffer(ptr); in DeallocateRaw()
86 const auto& buffer = sycl_device_->get_sycl_buffer(ptr); in RequestedSize()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d_sycl.h198 device.get_sycl_buffer(tensor_in.template flat<T>().data());
200 device.get_sycl_buffer(output->template flat<T>().data());
364 device.get_sycl_buffer(tensor_in.template flat<T>().data());
366 device.get_sycl_buffer(tensor_out.template flat<T>().data());
368 device.get_sycl_buffer(out_backprop.template flat<T>().data());
370 device.get_sycl_buffer(output->template flat<T>().data());
484 device.get_sycl_buffer(tensor_in.template flat<T>().data());
486 device.get_sycl_buffer(tensor_out.template flat<T>().data());
488 device.get_sycl_buffer(out_backprop.template flat<T>().data());
490 device.get_sycl_buffer(output->template flat<T>().data());
[all …]
Drandom_op.cc690 auto buffer = device.get_sycl_buffer(data); in operator ()()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceSycl.h70 …return (get_sycl_buffer<T>(num_bytes, ptr)->template get_access<AcMd, cl::sycl::access::target::gl… in get_sycl_accessor()
81 …template <typename T> inline cl::sycl::buffer<T, 1>* get_sycl_buffer(size_t num_bytes,const T * pt… in get_sycl_buffer() function
DTensorReductionSycl.h137 …auto out_buffer =dev.template get_sycl_buffer<typename Eigen::internal::remove_all<CoeffReturnType…