| /external/tensorflow/tensorflow/core/kernels/ |
| D | fill_functor.cc | 66 void SetZeroFunctor<Eigen::SyclDevice, T>::operator()( in operator ()()
67 const Eigen::SyclDevice& d, typename TTypes<T>::Flat out) { in operator ()()
72 template struct SetZeroFunctor<Eigen::SyclDevice, T>;
111 void SetOneFunctor<Eigen::SyclDevice, T>::operator()( in operator ()()
112 const Eigen::SyclDevice& d, typename TTypes<T>::Flat out) { in operator ()()
117 template struct SetOneFunctor<Eigen::SyclDevice, T>;
144 struct FillFunctor<Eigen::SyclDevice, T> {
145 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out, in operator ()()
159 #define DEFINE_FILL_SYCL(T) template struct FillFunctor<Eigen::SyclDevice, T>;
|
| D | split_lib_cpu.cc | 49 void Split<Eigen::SyclDevice, T, NDims>::operator()( in DEFINE_CPU_KERNELS()
50 const Eigen::SyclDevice& d, typename TTypes<T, NDims>::Tensor output, in DEFINE_CPU_KERNELS()
58 template struct Split<Eigen::SyclDevice, T, 2>; \
59 template struct Split<Eigen::SyclDevice, T, 3>;
|
| D | fill_functor.h | 51 struct SetZeroFunctor<Eigen::SyclDevice, T> {
52 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out);
78 struct SetOneFunctor<Eigen::SyclDevice, T> {
79 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out);
|
| D | slice_op_cpu_impl.h | 37 using SyclDevice = Eigen::SyclDevice; variable
40 template struct functor::Slice<SyclDevice, T, CPU_PROVIDED_IXDIM>;
|
| D | snapshot_op.cc | 65 typedef Eigen::SyclDevice SyclDevice; typedef
69 SnapshotOp<SyclDevice, TYPE>);
|
| D | split_lib.h | 53 struct Split<Eigen::SyclDevice, T> {
54 void operator()(const Eigen::SyclDevice& d,
|
| D | concat_lib_cpu.cc | 90 const Eigen::SyclDevice& d, in ConcatSYCL()
99 const Eigen::SyclDevice&, \
|
| D | inplace_ops.cc | 29 typedef Eigen::SyclDevice SyclDevice; typedef
65 Status DoParallelConcat(const SyclDevice& d, const Tensor& value, int32 loc, in DoParallelConcat()
71 return DoParallelConcatUpdate<SyclDevice, type>(d, value, loc, output); in DoParallelConcat()
183 ParallelConcatStart<SyclDevice, type>);
198 ParallelConcatUpdate<SyclDevice>);
|
| D | cast_op_impl_float.cc | 38 typedef Eigen::SyclDevice SYCLDevice;
|
| D | cast_op_impl_bool.cc | 36 typedef Eigen::SyclDevice SYCLDevice;
|
| D | cast_op_impl_double.cc | 36 typedef Eigen::SyclDevice SYCLDevice;
|
| D | cast_op_impl_int64.cc | 36 typedef Eigen::SyclDevice SYCLDevice;
|
| D | cast_op_impl_uint32.cc | 36 typedef Eigen::SyclDevice SYCLDevice;
|
| D | cast_op_impl_uint8.cc | 36 typedef Eigen::SyclDevice SYCLDevice;
|
| D | cast_op_impl_int16.cc | 36 typedef Eigen::SyclDevice SYCLDevice;
|
| /external/eigen/bench/tensors/ |
| D | 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/ |
| D | device_base.h | 34 struct SyclDevice;
177 void set_eigen_sycl_device(Eigen::SyclDevice* d) { eigen_sycl_device_ = d; } in set_eigen_sycl_device()
209 virtual const Eigen::SyclDevice* eigen_sycl_device() const { in eigen_sycl_device()
272 Eigen::SyclDevice* eigen_sycl_device_ = nullptr;
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_sycl.cpp | 25 static void test_full_reductions_sycl(const Eigen::SyclDevice& sycl_device) { in test_full_reductions_sycl()
55 static void test_first_dim_reductions_sycl(const Eigen::SyclDevice& sycl_device) { in test_first_dim_reductions_sycl()
93 static void test_last_dim_reductions_sycl(const Eigen::SyclDevice &sycl_device) { in test_last_dim_reductions_sycl()
133 Eigen::SyclDevice sycl_device(s); in test_cxx11_tensor_reduction_sycl()
|
| D | cxx11_tensor_broadcast_sycl.cpp | 24 using Eigen::SyclDevice;
28 static void test_broadcast_sycl(const Eigen::SyclDevice &sycl_device){ in test_broadcast_sycl()
72 Eigen::SyclDevice sycl_device(s); in test_cxx11_tensor_broadcast_sycl()
|
| D | cxx11_tensor_device_sycl.cpp | 23 void test_device_sycl(const Eigen::SyclDevice &sycl_device) { in test_device_sycl()
29 Eigen::SyclDevice sycl_device(s); in test_cxx11_tensor_device_sycl()
|
| D | cxx11_tensor_forced_eval_sycl.cpp | 25 void test_forced_eval_sycl(const Eigen::SyclDevice &sycl_device) { in test_forced_eval_sycl()
68 Eigen::SyclDevice sycl_device(s); in test_cxx11_tensor_forced_eval_sycl()
|
| D | cxx11_tensor_sycl.cpp | 26 using Eigen::SyclDevice;
30 void test_sycl_cpu(const Eigen::SyclDevice &sycl_device) { in test_sycl_cpu()
157 Eigen::SyclDevice sycl_device(s); in test_cxx11_tensor_sycl()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorReductionSycl.h | 30 static void run(BufferTOut* bufOut, BufferTIn& bufI, const Eigen::SyclDevice& dev, size_t length, s… in run()
103 struct FullReducer<Self, Op, const Eigen::SyclDevice, Vectorizable> {
108 …static void run(const Self& self, Op& reducer, const Eigen::SyclDevice& dev, CoeffReturnType* outp…
182 struct InnerReducer<Self, Op, const Eigen::SyclDevice> {
187 …static bool run(const Self& self, Op& reducer, const Eigen::SyclDevice& dev, CoeffReturnType* outp…
|
| D | TensorDeviceSycl.h | 19 struct SyclDevice { struct
28 template<typename dev_Selector> SyclDevice(dev_Selector s) in SyclDevice() argument
45 ~SyclDevice() { deallocate_all(); } in ~SyclDevice() argument
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.h | 55 Eigen::SyclDevice* getSyclDevice() { return sycl_device_; } in getSyclDevice()
67 Eigen::SyclDevice* sycl_device_ GUARDED_BY(mu_); // owned
|