Home
last modified time | relevance | path

Searched refs:TensorContractionOp (Results 1 – 5 of 5) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContraction.h25 struct traits<TensorContractionOp<Dimensions, LhsXprType, RhsXprType> >
50 struct eval<TensorContractionOp<Dimensions, LhsXprType, RhsXprType>, Eigen::Dense>
52 typedef const TensorContractionOp<Dimensions, LhsXprType, RhsXprType>& type;
56 struct nested<TensorContractionOp<Dimensions, LhsXprType, RhsXprType>, 1, typename eval<TensorContr…
58 typedef TensorContractionOp<Dimensions, LhsXprType, RhsXprType> type;
62 struct traits<TensorEvaluator<const TensorContractionOp<Indices_, LeftArgType_, RightArgType_>, Dev…
75 class TensorContractionOp : public TensorBase<TensorContractionOp<Indices, LhsXprType, RhsXprType>,…
78 typedef typename Eigen::internal::traits<TensorContractionOp>::Scalar Scalar;
81 typedef typename Eigen::internal::nested<TensorContractionOp>::type Nested;
82 typedef typename Eigen::internal::traits<TensorContractionOp>::StorageKind StorageKind;
[all …]
DTensorForwardDeclarations.h40 …mplate<typename Dimensions, typename LeftXprType, typename RightXprType> class TensorContractionOp; variable
DTensorContractionThreadPool.h60 struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, ThreadPoolDev…
61 …public TensorContractionEvaluatorBase<TensorEvaluator<const TensorContractionOp<Indices, LeftArgTy…
65 …typedef TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, Device> Sel…
68 typedef TensorContractionOp<Indices, LeftArgType, RightArgType> XprType;
DTensorContractionCuda.h1206 struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, GpuDevice> :
1207 …public TensorContractionEvaluatorBase<TensorEvaluator<const TensorContractionOp<Indices, LeftArgTy…
1211 …typedef TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, Device> Sel…
1214 typedef TensorContractionOp<Indices, LeftArgType, RightArgType> XprType;
DTensorBase.h482 const TensorContractionOp<const Dimensions, const Derived, const OtherDerived>
484 …return TensorContractionOp<const Dimensions, const Derived, const OtherDerived>(derived(), other.d… in contract()