Home
last modified time | relevance | path

Searched refs:TensorReshapingOp (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h60 TensorReshapingOp<
66 const Eigen::TensorForcedEvalOp<const TensorReshapingOp<
74 const TensorReshapingOp<
79 TensorReshapingOp<
85 const TensorReshapingOp<
90 const Eigen::TensorForcedEvalOp<const TensorReshapingOp<
328 TensorReshapingOp<
332 const TensorReshapingOp<
335 const TensorReshapingOp<
338 TensorReshapingOp<
[all …]
Deigen_pooling.h41 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<
398 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
408 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
516 EIGEN_ALWAYS_INLINE static const TensorReshapingOp<
520 const TensorReshapingOp<
[all …]
Deigen_backward_cuboid_convolutions.h49 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 …]
Deigen_spatial_convolutions-inl.h62 const TensorReshapingOp<NewDimension,
73 const TensorReshapingOp<
83 const TensorReshapingOp<
97 const TensorReshapingOp< in TensorContractionInputMapper()
274 const TensorReshapingOp<
517 const TensorReshapingOp<NewDimension,
529 const TensorReshapingOp<
539 const TensorReshapingOp<
812 const TensorReshapingOp<
821 const TensorReshapingOp<
[all …]
Deigen_cuboid_convolution.h64 TensorEvaluator<const TensorReshapingOp<NewDimension,
74 TensorEvaluator<const TensorReshapingOp<
83 TensorEvaluator<const TensorReshapingOp<
97 const TensorReshapingOp< in TensorContractionInputMapper()
341 TensorEvaluator<const TensorReshapingOp<
686 TensorEvaluator<const TensorReshapingOp<NewDimension,
698 TensorEvaluator<const TensorReshapingOp<
707 TensorEvaluator<const TensorReshapingOp<
1009 TensorEvaluator<const TensorReshapingOp<
1018 TensorEvaluator<const TensorReshapingOp<
[all …]
Deigen_spatial_convolutions.h40 const TensorReshapingOp<
49 const TensorReshapingOp<
Deigen_spatial_convolutions_test.cc1417 const TensorReshapingOp< in PackRhsHelper()
1495 TensorReshapingOp<NewDimension, decltype(image_patch_op)>( in PackRhsHelper()
1590 TensorEvaluator<const TensorReshapingOp<NewDimension, ArgType>, in PackLhsHelper()
1662 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorMorphing.h24 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 …]
DTensorBase.h737 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()
DTensorForwardDeclarations.h49 template<typename NewDimensions, typename XprType> class TensorReshapingOp; variable
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h72 TensorReshapingOp<
78 const TensorReshapingOp<
81 const TensorReshapingOp<
85 TensorReshapingOp<
91 const TensorReshapingOp<
94 const TensorReshapingOp<
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h69 TensorReshapingOp<
75 const TensorReshapingOp<
78 const TensorReshapingOp<
82 TensorReshapingOp<
88 const TensorReshapingOp<
91 const TensorReshapingOp<