Searched refs:TensorContractionOp (Results 1 – 11 of 11) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContraction.h | 25 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 …]
|
| D | TensorContractionThreadPool.h | 60 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;
|
| D | TensorForwardDeclarations.h | 40 …mplate<typename Dimensions, typename LeftXprType, typename RightXprType> class TensorContractionOp; variable
|
| D | TensorContractionCuda.h | 1206 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;
|
| D | TensorBase.h | 482 const TensorContractionOp<const Dimensions, const Derived, const OtherDerived>
484 …return TensorContractionOp<const Dimensions, const Derived, const OtherDerived>(derived(), other.d… in contract()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 63 const TensorContractionOp<
82 const TensorContractionOp<
330 const TensorContractionOp<
340 const TensorContractionOp<
|
| D | eigen_backward_cuboid_convolutions.h | 52 const TensorContractionOp<
71 const TensorContractionOp<
336 const TensorContractionOp<
367 const TensorContractionOp<
|
| D | eigen_spatial_convolutions-inl.h | 1335 const TensorContractionOp<
1348 const TensorContractionOp<
|
| D | eigen_cuboid_convolution.h | 1625 const TensorContractionOp<
1637 const TensorContractionOp<
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 75 const TensorContractionOp<
88 const TensorContractionOp<
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 72 const TensorContractionOp<
85 const TensorContractionOp<
|