Home
last modified time | relevance | path

Searched refs:SYCLDevice (Results 1 – 25 of 75) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dtranspose_functor_cpu.cc140 typedef Eigen::SyclDevice SYCLDevice; typedef
144 void TransposeSYCL(const SYCLDevice& d, const Tensor& in, in TransposeSYCL()
149 TransposeUsingEigen<SYCLDevice, T, 1>(d, in, perm, conjugate, out); in TransposeSYCL()
152 TransposeUsingEigen<SYCLDevice, T, 2>(d, in, perm, conjugate, out); in TransposeSYCL()
155 TransposeUsingEigen<SYCLDevice, T, 3>(d, in, perm, conjugate, out); in TransposeSYCL()
158 TransposeUsingEigen<SYCLDevice, T, 4>(d, in, perm, conjugate, out); in TransposeSYCL()
161 TransposeUsingEigen<SYCLDevice, T, 5>(d, in, perm, conjugate, out); in TransposeSYCL()
164 TransposeUsingEigen<SYCLDevice, T, 6>(d, in, perm, conjugate, out); in TransposeSYCL()
167 TransposeUsingEigen<SYCLDevice, T, 7>(d, in, perm, conjugate, out); in TransposeSYCL()
170 TransposeUsingEigen<SYCLDevice, T, 8>(d, in, perm, conjugate, out); in TransposeSYCL()
[all …]
Daggregate_ops_cpu.h27 typedef Eigen::SyclDevice SYCLDevice; typedef
144 struct Add2Functor<SYCLDevice, T> {
145 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat out,
148 Add2EigenImpl<SYCLDevice, T>::Compute(d, out, in1, in2);
152 struct Add3Functor<SYCLDevice, T> {
153 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat out,
157 Add3EigenImpl<SYCLDevice, T>::Compute(d, out, in1, in2, in3);
161 struct Add4Functor<SYCLDevice, T> {
162 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat out,
167 Add4EigenImpl<SYCLDevice, T>::Compute(d, out, in1, in2, in3, in4);
[all …]
Dcwise_ops_sycl_common.h32 typedef Eigen::SyclDevice SYCLDevice; typedef
35 void Assign(const SYCLDevice& d, OUT out, RHS rhs) { in Assign()
41 struct UnaryFunctor<SYCLDevice, Functor> {
42 void operator()(const SYCLDevice& d, typename Functor::tout_type out,
50 struct BinaryFunctor<SYCLDevice, Functor, NDIMS, has_errors> {
51 void operator()(const SYCLDevice& d, typename Functor::tout_type out,
58 void Left(const SYCLDevice& d, typename Functor::tout_type out,
70 void Right(const SYCLDevice& d, typename Functor::tout_type out,
81 void BCast(const SYCLDevice& d,
112 #define DEFINE_UNARY1(F, T) template struct UnaryFunctor<SYCLDevice, F<T> >
[all …]
Ddense_update_functor.h31 typedef Eigen::SyclDevice SYCLDevice; typedef
70 struct DenseUpdate<SYCLDevice, T, ADD> {
71 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat params,
78 struct DenseUpdate<SYCLDevice, T, SUB> {
79 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat params,
86 struct DenseUpdate<SYCLDevice, T, ASSIGN> {
87 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat params,
Drelu_op.cc33 typedef Eigen::SyclDevice SYCLDevice; typedef
219 ReluOp<SYCLDevice, type>); \
222 ReluGradOp<SYCLDevice, type>); \
225 Relu6Op<SYCLDevice, type>); \
228 Relu6GradOp<SYCLDevice, type>); \
231 LeakyReluOp<SYCLDevice, type>); \
234 LeakyReluGradOp<SYCLDevice, type>); \
237 EluOp<SYCLDevice, type>); \
240 EluGradOp<SYCLDevice, type>); \
243 SeluOp<SYCLDevice, type>); \
[all …]
Dsoftmax_op.cc33 typedef Eigen::SyclDevice SYCLDevice; typedef
51 struct SoftmaxFunctor<SYCLDevice, T> : SoftmaxFunctorBase<SYCLDevice, T> {};
101 SoftmaxOp<SYCLDevice, float>);
104 SoftmaxOp<SYCLDevice, double>);
Dcwise_op_select.cc33 typedef Eigen::SyclDevice SYCLDevice; typedef
180 SelectOp<SYCLDevice, type>);
206 struct SelectFunctor<SYCLDevice, T> : SelectFunctorBase<SYCLDevice, T> {};
254 struct SelectScalarFunctor<SYCLDevice, T>
255 : SelectScalarFunctorBase<SYCLDevice, T> {};
329 struct BatchSelectFunctor<SYCLDevice, T>
330 : BatchSelectFunctorBase<SYCLDevice, T> {};
Dtile_ops_cpu_impl.h49 typedef Eigen::SyclDevice SYCLDevice; typedef
53 template struct TileGrad<SYCLDevice, T, NDIM>; \
54 template struct ReduceAndReshape<SYCLDevice, T, NDIM, 1>;
Drandom_op.h58 typedef Eigen::SyclDevice SYCLDevice;
61 struct FillPhiloxRandom<SYCLDevice, Distribution> {
62 void operator()(OpKernelContext* ctx, const SYCLDevice& d,
Ddense_update_ops.cc91 typedef Eigen::SyclDevice SYCLDevice; typedef
122 AssignOpT<SYCLDevice, type>);
156 DenseUpdateOp<SYCLDevice, type, DenseUpdateType::ADD>); \
159 DenseUpdateOp<SYCLDevice, type, DenseUpdateType::SUB>);
Dxent_op.cc34 typedef Eigen::SyclDevice SYCLDevice; typedef
124 struct XentFunctor<SYCLDevice, T> : XentFunctorBase<SYCLDevice, T> {};
156 SoftmaxXentWithLogitsOp<SYCLDevice, float>);
Dtile_functor_cpu.cc74 typedef Eigen::SyclDevice SYCLDevice; typedef
77 template struct Tile<SYCLDevice, T, int32>; \
78 template struct Tile<SYCLDevice, T, int64>;
Dpooling_ops_3d_sycl.h27 typedef Eigen::SyclDevice SYCLDevice; typedef
178 struct LaunchPoolingOp<SYCLDevice, T, MAX> {
185 const SYCLDevice& device = context->eigen_device<SYCLDevice>();
346 struct LaunchMaxPooling3dGradOp<SYCLDevice, T> {
354 const SYCLDevice& device = context->eigen_device<SYCLDevice>();
475 struct LaunchMaxPooling3dGradGradOp<SYCLDevice, T> {
479 const SYCLDevice& device = context->eigen_device<SYCLDevice>();
584 struct LaunchPoolingOp<SYCLDevice, T, AVG> {
591 const SYCLDevice& device = context->eigen_device<SYCLDevice>();
717 struct LaunchAvgPooling3dGradOp<SYCLDevice, T> {
[all …]
Dcast_op_impl_float.cc38 typedef Eigen::SyclDevice SYCLDevice; typedef
40 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, float); in GetSyclCastFromFloat()
Dcast_op_impl_bool.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, bool); in GetSyclCastFromBool()
Dcast_op_impl_double.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, double); in GetSyclCastFromDouble()
Dcast_op_impl_int64.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, int64); in GetSyclCastFromInt64()
Dcast_op_impl_uint32.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, uint32); in GetSyclCastFromUint32()
Dcast_op_impl_uint8.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, uint8); in GetSyclCastFromUint8()
Dcast_op_impl_int16.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, int16); in GetSyclCastFromInt16()
Dcast_op_impl_uint16.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, uint16); in GetSyclCastFromUint16()
Dcast_op_impl_int8.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, int8); in GetSyclCastFromInt8()
Dcast_op_impl_uint64.cc36 typedef Eigen::SyclDevice SYCLDevice; typedef
38 CURRY_TYPES3_NO_HALF(CAST_CASE, SYCLDevice, uint64); in GetSyclCastFromUint64()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.cc26 SYCLDevice::~SYCLDevice() {} in ~SYCLDevice()
28 void SYCLDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
38 Allocator* SYCLDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()
45 Status SYCLDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
77 Status SYCLDevice::FillContextMap(const Graph* graph, in FillContextMap()
90 Status SYCLDevice::Sync() { in Sync()
/external/tensorflow/tensorflow/core/framework/
Dregister_types_traits.h25 typedef Eigen::SyclDevice SYCLDevice; typedef
79 struct proxy_type_pod<SYCLDevice, 8> {
83 struct proxy_type_pod<SYCLDevice, 4> {

123