Searched refs:TensorReshapingOp (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 69 TensorReshapingOp<
75 const TensorReshapingOp<
78 const TensorReshapingOp<
82 TensorReshapingOp<
88 const TensorReshapingOp<
91 const TensorReshapingOp<
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorMorphing.h | 24 struct traits<TensorReshapingOp<NewDimensions, XprType> > : public traits<XprType>
37 struct eval<TensorReshapingOp<NewDimensions, XprType>, Eigen::Dense>
39 typedef const TensorReshapingOp<NewDimensions, XprType>& type;
43 struct nested<TensorReshapingOp<NewDimensions, XprType>, 1, typename eval<TensorReshapingOp<NewDime…
45 typedef TensorReshapingOp<NewDimensions, XprType> type;
53 class TensorReshapingOp : public TensorBase<TensorReshapingOp<NewDimensions, XprType>, WriteAccesso…
56 typedef typename Eigen::internal::traits<TensorReshapingOp>::Scalar Scalar;
58 typedef typename Eigen::internal::nested<TensorReshapingOp>::type Nested;
59 typedef typename Eigen::internal::traits<TensorReshapingOp>::StorageKind StorageKind;
60 typedef typename Eigen::internal::traits<TensorReshapingOp>::Index Index;
[all …]
|
| D | TensorBase.h | 737 const TensorReshapingOp<const NewDimensions, const Derived>
739 return TensorReshapingOp<const NewDimensions, const Derived>(derived(), newDimensions); in reshape()
906 const TensorReshapingOp<const NewDimensions, const Derived>
908 return TensorReshapingOp<const NewDimensions, const Derived>(derived(), newDimensions); in reshape()
911 TensorReshapingOp<const NewDimensions, Derived>
913 return TensorReshapingOp<const NewDimensions, Derived>(derived(), newDimensions); in reshape()
|
| D | TensorForwardDeclarations.h | 49 template<typename NewDimensions, typename XprType> class TensorReshapingOp; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 60 TensorReshapingOp<
66 const TensorReshapingOp<
74 const TensorReshapingOp<
79 TensorReshapingOp<
86 const TensorReshapingOp<
91 const TensorReshapingOp<
341 const Eigen::TensorReshapingOp<
347 const TensorReshapingOp<
356 const TensorReshapingOp<
373 const Eigen::TensorReshapingOp<
[all …]
|
| D | eigen_pooling.h | 41 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
51 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
160 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
164 const TensorReshapingOp<
170 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
175 const TensorReshapingOp<
401 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
411 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
519 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
523 const TensorReshapingOp<
[all …]
|
| D | eigen_backward_cuboid_convolutions.h | 49 TensorReshapingOp<
55 const Eigen::TensorForcedEvalOp<const TensorReshapingOp<
63 const TensorReshapingOp<
68 TensorReshapingOp<
74 const TensorReshapingOp<
79 const Eigen::TensorForcedEvalOp<const TensorReshapingOp<
333 const Eigen::TensorReshapingOp<
339 const Eigen::TensorForcedEvalOp<const TensorReshapingOp<
347 const TensorReshapingOp<
364 const Eigen::TensorReshapingOp<
[all …]
|
| D | eigen_cuboid_convolution.h | 66 TensorEvaluator<const TensorReshapingOp<NewDimension,
76 TensorEvaluator<const TensorReshapingOp<
85 TensorEvaluator<const TensorReshapingOp<
99 const TensorReshapingOp< in TensorContractionInputMapper()
343 TensorEvaluator<const TensorReshapingOp<
879 TensorEvaluator<const TensorReshapingOp<NewDimension,
891 TensorEvaluator<const TensorReshapingOp<
900 TensorEvaluator<const TensorReshapingOp<
1203 TensorEvaluator<const TensorReshapingOp<
1212 TensorEvaluator<const TensorReshapingOp<
[all …]
|
| D | eigen_spatial_convolutions-inl.h | 64 const TensorReshapingOp<NewDimension,
75 const TensorReshapingOp<
85 const TensorReshapingOp<
101 const TensorReshapingOp< in TensorContractionInputMapper()
280 const TensorReshapingOp<
682 const TensorReshapingOp<NewDimension,
694 const TensorReshapingOp<
704 const TensorReshapingOp<
1049 const TensorReshapingOp<
1058 const TensorReshapingOp<
[all …]
|
| D | eigen_spatial_convolutions.h | 86 const TensorReshapingOp<
95 const TensorReshapingOp<
|
| D | eigen_spatial_convolutions_test.cc | 1421 const TensorReshapingOp< in PackRhsHelper()
1519 TensorReshapingOp<NewDimension, decltype(image_patch_op)>( in PackRhsHelper()
1620 TensorEvaluator<const TensorReshapingOp<NewDimension, ArgType>, in PackLhsHelper()
1696 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()
|
| D | eigen_contraction_kernel.h | 470 const TensorReshapingOp< \
|