Lines Matching refs:Axis
23 template<typename Axis, typename LhsXprType, typename RhsXprType>
24 struct traits<TensorConcatenationOp<Axis, LhsXprType, RhsXprType> >
42 template<typename Axis, typename LhsXprType, typename RhsXprType>
43 struct eval<TensorConcatenationOp<Axis, LhsXprType, RhsXprType>, Eigen::Dense>
45 typedef const TensorConcatenationOp<Axis, LhsXprType, RhsXprType>& type;
48 template<typename Axis, typename LhsXprType, typename RhsXprType>
49 struct nested<TensorConcatenationOp<Axis, LhsXprType, RhsXprType>, 1, typename eval<TensorConcatenationOp<Axis, LhsXprType, RhsXprType> >::type>
51 typedef TensorConcatenationOp<Axis, LhsXprType, RhsXprType> type;
57 template<typename Axis, typename LhsXprType, typename RhsXprType>
58 class TensorConcatenationOp : public TensorBase<TensorConcatenationOp<Axis, LhsXprType, RhsXprType>, WriteAccessors>
69 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorConcatenationOp(const LhsXprType& lhs, const RhsXprType& rhs, Axis axis)
70 : m_lhs_xpr(lhs), m_rhs_xpr(rhs), m_axis(axis) {}
80 EIGEN_DEVICE_FUNC const Axis& axis() const { return m_axis; }
104 const Axis m_axis;
109 template<typename Axis, typename LeftArgType, typename RightArgType, typename Device>
110 struct TensorEvaluator<const TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device>
112 typedef TensorConcatenationOp<Axis, LeftArgType, RightArgType> XprType;
128 : m_leftImpl(op.lhsExpression(), device), m_rightImpl(op.rhsExpression(), device), m_axis(op.axis())
287 const Axis m_axis;
291 template<typename Axis, typename LeftArgType, typename RightArgType, typename Device>
292 struct TensorEvaluator<TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device>
293 : public TensorEvaluator<const TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device>
295 typedef TensorEvaluator<const TensorConcatenationOp<Axis, LeftArgType, RightArgType>, Device> Base;
296 typedef TensorConcatenationOp<Axis, LeftArgType, RightArgType> XprType;
