HomeSort by relevance Sort by last modified time
    Searched refs:SyclDevice (Results 1 - 25 of 85) sorted by null

1 2 3 4

  /external/tensorflow/tensorflow/core/kernels/
slice_op_cpu_impl.h 37 using SyclDevice = Eigen::SyclDevice;
40 template struct functor::Slice<SyclDevice, T, CPU_PROVIDED_IXDIM>;
snapshot_op.cc 35 typedef Eigen::SyclDevice SyclDevice;
39 SnapshotOp<SyclDevice, TYPE>);
fill_functor.h 49 // Partial specialization of SetZeroFunctor<Device=Eigen::SyclDevice, T>.
51 struct SetZeroFunctor<Eigen::SyclDevice, T> {
52 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out);
76 // Partial specialization of SetOneFunctor<Device=Eigen::SyclDevice, T>.
78 struct SetOneFunctor<Eigen::SyclDevice, T> {
79 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out);
fill_functor.cc 61 void SetZeroFunctor<Eigen::SyclDevice, T>::operator()(
62 const Eigen::SyclDevice& d, typename TTypes<T>::Flat out) {
67 template struct SetZeroFunctor<Eigen::SyclDevice, T>;
106 void SetOneFunctor<Eigen::SyclDevice, T>::operator()(
107 const Eigen::SyclDevice& d, typename TTypes<T>::Flat out) {
112 template struct SetOneFunctor<Eigen::SyclDevice, T>;
139 struct FillFunctor<Eigen::SyclDevice, T> {
140 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out,
154 #define DEFINE_FILL_SYCL(T) template struct FillFunctor<Eigen::SyclDevice, T>;
split_lib_cpu.cc 47 void Split<Eigen::SyclDevice, T>::operator()(
48 const Eigen::SyclDevice& d, typename TTypes<T, 3>::Tensor output,
55 #define DEFINE_SYCL_KERNELS(T) template struct Split<Eigen::SyclDevice, T>;
cast_op_impl_bool.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, bool);
cast_op_impl_double.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, double);
cast_op_impl_int16.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, int16);
cast_op_impl_int32.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, int32);
cast_op_impl_int64.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, int64);
cast_op_impl_int8.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, int8);
cast_op_impl_uint16.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, uint16);
cast_op_impl_uint8.cc 38 typedef Eigen::SyclDevice SYCLDevice;
41 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, uint8);
split_lib.h 53 struct Split<Eigen::SyclDevice, T> {
54 void operator()(const Eigen::SyclDevice& d,
concat_lib.h 62 const Eigen::SyclDevice& d,
random_op.h 58 typedef Eigen::SyclDevice SYCLDevice;
61 struct FillPhiloxRandom<SYCLDevice, Distribution> {
62 void operator()(OpKernelContext* ctx, const SYCLDevice& d,
inplace_ops.cc 29 typedef Eigen::SyclDevice SyclDevice;
64 Status DoParallelConcat(const SyclDevice& d, const Tensor& value, int32 loc,
70 return DoParallelConcatUpdate<SyclDevice, type>(d, value, loc, output);
181 ParallelConcatStart<SyclDevice, type>);
196 ParallelConcatUpdate<SyclDevice>);
  /external/eigen/unsupported/test/
cxx11_tensor_device_sycl.cpp 23 void test_device_sycl(const Eigen::SyclDevice &sycl_device) {
29 Eigen::SyclDevice sycl_device(s);
cxx11_tensor_broadcast_sycl.cpp 24 using Eigen::SyclDevice;
28 static void test_broadcast_sycl(const Eigen::SyclDevice &sycl_device){
72 Eigen::SyclDevice sycl_device(s);
cxx11_tensor_reduction_sycl.cpp 25 static void test_full_reductions_sycl(const Eigen::SyclDevice& sycl_device) {
55 static void test_first_dim_reductions_sycl(const Eigen::SyclDevice& sycl_device) {
93 static void test_last_dim_reductions_sycl(const Eigen::SyclDevice &sycl_device) {
133 Eigen::SyclDevice sycl_device(s);
cxx11_tensor_forced_eval_sycl.cpp 25 void test_forced_eval_sycl(const Eigen::SyclDevice &sycl_device) {
68 Eigen::SyclDevice sycl_device(s);
  /external/eigen/bench/tensors/
tensor_benchmarks_sycl.cc 9 using Eigen::SyclDevice;
30 Eigen::SyclDevice device(q); \
31 BenchmarkSuite<Eigen::SyclDevice, float> suite(device, N); \
  /external/tensorflow/tensorflow/core/framework/
device_base.h 33 struct SyclDevice;
160 void set_eigen_sycl_device(Eigen::SyclDevice* d) { eigen_sycl_device_ = d; }
186 virtual const Eigen::SyclDevice* eigen_sycl_device() const {
237 Eigen::SyclDevice* eigen_sycl_device_ = nullptr;
  /external/tensorflow/tensorflow/core/common_runtime/sycl/
sycl_allocator.h 55 Eigen::SyclDevice* getSyclDevice() { return sycl_device_; }
67 Eigen::SyclDevice* sycl_device_ GUARDED_BY(mu_); // owned
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorDeviceSycl.h 19 struct SyclDevice {
24 /// per pointer. The lifespan of the buffer now depends on the lifespan of SyclDevice.
28 template<typename dev_Selector> SyclDevice(dev_Selector s)
45 ~SyclDevice() { deallocate_all(); }

Completed in 364 milliseconds

1 2 3 4