Searched refs:TensorReductionOp (Results 1 – 12 of 12) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 525 const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived>
527 …return TensorReductionOp<internal::SumReducer<CoeffReturnType>, const Dims, const Derived>(derived… in sum()
530 …const TensorReductionOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDimen…
533 …return TensorReductionOp<internal::SumReducer<CoeffReturnType>, const DimensionList<Index, NumDime… in sum()
537 const TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const Dims, const Derived>
539 …return TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const Dims, const Derived>(derive… in mean()
542 …const TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDime…
545 …return TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDim… in mean()
549 const TensorReductionOp<internal::ProdReducer<CoeffReturnType>, const Dims, const Derived>
551 …return TensorReductionOp<internal::ProdReducer<CoeffReturnType>, const Dims, const Derived>(derive… in prod()
[all …]
|
| D | TensorReduction.h | 25 struct traits<TensorReductionOp<Op, Dims, XprType, MakePointer_> >
44 struct eval<TensorReductionOp<Op, Dims, XprType, MakePointer_>, Eigen::Dense>
46 typedef const TensorReductionOp<Op, Dims, XprType, MakePointer_>& type;
50 struct nested<TensorReductionOp<Op, Dims, XprType, MakePointer_>, 1, typename eval<TensorReductionO…
52 typedef TensorReductionOp<Op, Dims, XprType, MakePointer_> type;
350 class TensorReductionOp : public TensorBase<TensorReductionOp<Op, Dims, XprType, MakePointer_>, Rea…
352 typedef typename Eigen::internal::traits<TensorReductionOp>::Scalar Scalar;
355 typedef typename Eigen::internal::nested<TensorReductionOp>::type Nested;
356 typedef typename Eigen::internal::traits<TensorReductionOp>::StorageKind StorageKind;
357 typedef typename Eigen::internal::traits<TensorReductionOp>::Index Index;
[all …]
|
| D | TensorSyclExtractFunctors.h | 155 struct FunctorExtractor<TensorEvaluator<const TensorReductionOp<Op, Dims, ArgType, MakePointer_>, D…
156 …typedef TensorEvaluator<const TensorReductionOp<Op, Dims, ArgType, MakePointer_>, Device> Evaluato…
160 …FunctorExtractor(const TensorEvaluator<const TensorReductionOp<Op, Dims, ArgType, MakePointer_>, D…
166 struct FunctorExtractor<TensorEvaluator<TensorReductionOp<Op, Dims, ArgType, MakePointer_>, Device>>
167 : FunctorExtractor<TensorEvaluator<const TensorReductionOp<Op, Dims, ArgType, MakePointer_>, Device…
|
| D | TensorSyclLeafCount.h | 98 struct LeafCount<const TensorReductionOp<OP, Dim, Expr> > {
104 struct LeafCount<TensorReductionOp<OP, Dim, Expr> >: LeafCount<const TensorReductionOp<OP, Dim, Exp…
|
| D | TensorSyclExprConstructor.h | 212 struct ExprConstructor<CVQual TensorReductionOp<OP, Dim, OrigExpr, MakeGlobalPointer>,\
213 CVQual PlaceHolder<CVQual TensorReductionOp<OP, Dim, DevExpr>, N>, Params...> {\
214 … NumIndices= ValueCondition< TensorReductionOp<OP, Dim, DevExpr, MakeGlobalPointer>::NumDimensions…
215 …typedef CVQual TensorMap<Tensor<typename TensorReductionOp<OP, Dim, DevExpr, MakeGlobalPointer>::S…
216 NumIndices, 0, typename TensorReductionOp<OP, Dim, DevExpr>::Index>, 0, MakeGlobalPointer> Type;\
|
| D | TensorSyclConvertToDeviceExpression.h | 109 struct ConvertToDeviceExpression<CVQual TensorReductionOp<OP, Dim, subExpr, MakePointer_> > {\
110 …typedef CVQual TensorReductionOp<OP, Dim, typename ConvertToDeviceExpression<subExpr>::Type, MakeG…
|
| D | TensorSyclExtractAccessor.h | 182 struct ExtractAccessor<TensorEvaluator<const TensorReductionOp<OP, Dim, Expr>, Dev> > {
183 …static inline auto getTuple(cl::sycl::handler& cgh, const TensorEvaluator<const TensorReductionOp<…
191 struct ExtractAccessor<TensorEvaluator<TensorReductionOp<OP, Dim, Expr>, Dev> >
192 : ExtractAccessor<TensorEvaluator<const TensorReductionOp<OP, Dim, Expr>, Dev> >{};
|
| D | TensorArgMax.h | 208 …typedef typename TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<…
217 …Layout = TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType>…
290 …TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType> >, Devic…
|
| D | TensorSyclPlaceHolderExpr.h | 163 struct PlaceHolderExpression<CVQual TensorReductionOp<OP, Dims, Expr>, N>{\
164 typedef CVQual PlaceHolder<CVQual TensorReductionOp<OP, Dims,Expr>, N> Type;\
|
| D | TensorReductionSycl.h | 156 …const auto device_self_expr= TensorReductionOp<Op, Dims, decltype(device_expr.expr) ,MakeGlobalPoi…
219 …const auto device_self_expr= TensorReductionOp<Op, Dims, decltype(device_expr.expr) ,MakeGlobalPoi…
|
| D | TensorForwardDeclarations.h | 36 …ims, typename XprType, template <class> class MakePointer_ = MakePointer > class TensorReductionOp; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_pooling.h | 44 const TensorReductionOp<
54 const TensorReductionOp<
162 const TensorReductionOp<
172 const TensorReductionOp<
401 const TensorReductionOp<
411 const TensorReductionOp<
518 const TensorReductionOp<
528 const TensorReductionOp<
|