aboutsummaryrefslogtreecommitdiffhomepage
path: root/unsupported/Eigen/CXX11/src/Tensor/TensorSyclConvertToDeviceExpression.h
diff options
context:
space:
mode:
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorSyclConvertToDeviceExpression.h')
-rw-r--r--unsupported/Eigen/CXX11/src/Tensor/TensorSyclConvertToDeviceExpression.h14
1 files changed, 14 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorSyclConvertToDeviceExpression.h b/unsupported/Eigen/CXX11/src/Tensor/TensorSyclConvertToDeviceExpression.h
index 0336c9866..cc13ca963 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorSyclConvertToDeviceExpression.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorSyclConvertToDeviceExpression.h
@@ -124,6 +124,20 @@ KERNELBROKERCONVERTSLICEOP(const)
KERNELBROKERCONVERTSLICEOP()
#undef KERNELBROKERCONVERTSLICEOP
+
+#define KERNELBROKERCONVERTRESHAPEANDSHUFFLEOP(OPEXPR, CVQual)\
+template<typename Param, typename XprType>\
+struct ConvertToDeviceExpression<CVQual OPEXPR <Param, XprType> >{\
+ typedef CVQual OPEXPR<Param, typename ConvertToDeviceExpression<XprType>::Type> Type;\
+};
+
+KERNELBROKERCONVERTRESHAPEANDSHUFFLEOP(TensorReshapingOp, const)
+KERNELBROKERCONVERTRESHAPEANDSHUFFLEOP(TensorReshapingOp, )
+
+KERNELBROKERCONVERTRESHAPEANDSHUFFLEOP(TensorShufflingOp, const)
+KERNELBROKERCONVERTRESHAPEANDSHUFFLEOP(TensorShufflingOp, )
+#undef KERNELBROKERCONVERTRESHAPEANDSHUFFLEOP
+
} // namespace internal
} // namespace TensorSycl
} // namespace Eigen