Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_contraction_kernel.h248 typename LhsMapper::SubMapper, ColMajor>;
251 typename RhsMapper::SubMapper, ColMajor>;
257 gemm_pack_lhs<Scalar, StorageIndex, typename LhsMapper::SubMapper,
261 gemm_pack_rhs<Scalar, StorageIndex, typename RhsMapper::SubMapper,
269 Scalar* lhsBlock, const typename LhsMapper::SubMapper& data_mapper,
280 Scalar* rhsBlock, const typename RhsMapper::SubMapper& data_mapper,
Deigen_cuboid_convolution.h89 SubMapper; typedef
90 typedef SubMapper VectorMapper;
91 typedef SubMapper LinearMapper;
282 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const { in getSubMapper()
283 return SubMapper(*this, i, j); in getSubMapper()
1024 SubMapper;
1026 typedef SubMapper DataMapper;
1043 const SubMapper dm0 = rhs.getLinearMapper(0, j2 + 0);
1044 const SubMapper dm1 = rhs.getLinearMapper(0, j2 + 1);
1045 const SubMapper dm2 = rhs.getLinearMapper(0, j2 + 2);
[all …]
Deigen_spatial_convolutions-inl.h88 SubMapper; typedef
90 typedef SubMapper VectorMapper;
91 typedef SubMapper LinearMapper;
215 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const { in getSubMapper()
216 return SubMapper(*this, i, j); in getSubMapper()
826 SubMapper;
827 typedef SubMapper DataMapper;
844 const SubMapper dm0 = rhs.getLinearMapper(0, j2 + 0);
845 const SubMapper dm1 = rhs.getLinearMapper(0, j2 + 1);
846 const SubMapper dm2 = rhs.getLinearMapper(0, j2 + 2);
[all …]
Deigen_spatial_convolutions.h54 SubMapper;
56 typedef SubMapper DataMapper;
79 SubMapper lm = rhs.getLinearMapper(0, col);
108 SubMapper lm = rhs.getLinearMapper(0, col);
Deigen_spatial_convolutions_test.cc1429 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackRhsHelper() typedef
1439 Eigen::internal::gemm_pack_colmajor_block<float, Eigen::Index, SubMapper, in PackRhsHelper()
1443 Eigen::internal::gemm_pack_rhs<float, Eigen::Index, SubMapper, // in PackRhsHelper()
1541 SubMapper sub_mapper = in PackRhsHelper()
1601 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackLhsHelper() typedef
1611 Eigen::internal::gemm_pack_colmajor_block<float, Eigen::Index, SubMapper, in PackLhsHelper()
1616 Eigen::internal::gemm_pack_lhs<float, Eigen::Index, SubMapper, // in PackLhsHelper()
1700 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;