Home
last modified time | relevance | path

Searched refs:SubMapper (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_cuboid_convolution.h91 SubMapper; typedef
92 typedef SubMapper VectorMapper;
93 typedef SubMapper LinearMapper;
284 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const { in getSubMapper()
285 return SubMapper(*this, i, j); in getSubMapper()
1218 SubMapper;
1220 typedef SubMapper DataMapper;
1237 const SubMapper dm0 = rhs.getLinearMapper(0, j2 + 0);
1238 const SubMapper dm1 = rhs.getLinearMapper(0, j2 + 1);
1239 const SubMapper dm2 = rhs.getLinearMapper(0, j2 + 2);
[all …]
Deigen_spatial_convolutions-inl.h90 SubMapper; typedef
92 typedef SubMapper VectorMapper;
93 typedef SubMapper LinearMapper;
221 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const { in getSubMapper()
222 return SubMapper(*this, i, j); in getSubMapper()
1063 SubMapper;
1064 typedef SubMapper DataMapper;
1081 const SubMapper dm0 = rhs.getLinearMapper(0, j2 + 0);
1082 const SubMapper dm1 = rhs.getLinearMapper(0, j2 + 1);
1083 const SubMapper dm2 = rhs.getLinearMapper(0, j2 + 2);
[all …]
Deigen_contraction_kernel.h400 static bool block(const typename DataMapper::SubMapper& data_mapper,
433 static bool block(const typename DataMapper::SubMapper& data_mapper, \
563 typename LhsMapper::SubMapper, ColMajor>; \
566 typename RhsMapper::SubMapper, ColMajor>; \
579 gemm_pack_lhs<LhsScalar, StorageIndex, typename LhsMapper::SubMapper, \
583 gemm_pack_rhs<RhsScalar, StorageIndex, typename RhsMapper::SubMapper, \
631 LhsBlock* lhsBlock, const typename LhsMapper::SubMapper& data_mapper, \
651 RhsBlock* rhsBlock, const typename RhsMapper::SubMapper& data_mapper, \
772 typename LhsMapper::SubMapper, ColMajor>; \
775 typename RhsMapper::SubMapper, ColMajor>; \
[all …]
Deigen_spatial_convolutions.h100 SubMapper;
102 typedef SubMapper DataMapper;
186 SubMapper lm = rhs.getLinearMapper(0, col);
371 SubMapper lm = rhs.getLinearMapper(0, col);
Deigen_spatial_convolutions_test.cc1430 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackRhsHelper() typedef
1440 Eigen::internal::gemm_pack_colmajor_block<T, Eigen::Index, SubMapper, in PackRhsHelper()
1445 Eigen::internal::gemm_pack_rhs<T, Eigen::Index, SubMapper, // in PackRhsHelper()
1566 SubMapper sub_mapper = in PackRhsHelper()
1624 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackLhsHelper() typedef
1634 Eigen::internal::gemm_pack_colmajor_block<T, Eigen::Index, SubMapper, in PackLhsHelper()
1639 Eigen::internal::gemm_pack_lhs<T, Eigen::Index, SubMapper, // in PackLhsHelper()
1726 SubMapper sub_mapper = in PackLhsHelper()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionMapper.h442 …ntract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment> SubMapper;
443 typedef SubMapper VectorMapper;
453 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const {
454 return SubMapper(*this, i, j);
DTensorContractionThreadPool.h140 typename LhsMapper::SubMapper, Traits::mr,
144 RhsScalar, Index, typename RhsMapper::SubMapper, Traits::nr, ColMajor>
775 typedef internal::gemm_pack_lhs<LhsScalar, Index, typename LhsMapper::SubMapper, Traits::mr,
777 …typedef internal::gemm_pack_rhs<RhsScalar, Index, typename RhsMapper::SubMapper, Traits::nr, ColMa…
DTensorContraction.h466 …internal::gemm_pack_lhs<LhsScalar, Index, typename LhsMapper::SubMapper, mr, Traits::LhsProgress, …
467 internal::gemm_pack_rhs<RhsScalar, Index, typename RhsMapper::SubMapper, nr, ColMajor> pack_rhs;