| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorContractionBlocking.h | 29 typedef typename RhsMapper::Scalar RhsScalar;
35 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, mc_, nc_, num_threads);
38 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, nc_, mc_, num_threads);
|
| TensorContraction.h | 388 typedef typename internal::remove_const<typename EvalRightArgType::Scalar>::type RhsScalar;
401 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
418 internal::general_matrix_vector_product<Index,LhsScalar,LhsMapper,ColMajor,false,RhsScalar,RhsMapper,false>::run(
439 typedef typename internal::remove_const<typename EvalRightArgType::Scalar>::type RhsScalar;
440 typedef typename internal::gebp_traits<LhsScalar, RhsScalar> Traits;
457 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
467 internal::gemm_pack_rhs<RhsScalar, Index, typename RhsMapper::SubMapper, nr, ColMajor> pack_rhs;
469 internal::gebp_kernel<LhsScalar, RhsScalar, Index, OutputMapper, mr, nr, false, false> gebp;
489 RhsScalar* blockB = static_cast<RhsScalar *>(this->m_device.allocate(sizeB * sizeof(RhsScalar)))
[all...] |
| TensorContractionCuda.h | [all...] |
| TensorContractionThreadPool.h | 31 template<typename LhsScalar, typename RhsScalar, typename RhsMapper, typename OutputMapper, typename Index>
34 RhsScalar* blockB;
106 typedef typename internal::remove_const<typename EvalRightArgType::Scalar>::type RhsScalar;
107 typedef typename internal::gebp_traits<LhsScalar, RhsScalar> Traits;
124 RhsScalar;
125 typedef typename internal::gebp_traits<LhsScalar, RhsScalar> Traits;
134 RhsScalar, Index, internal::Rhs, RightEvaluator, right_nocontract_t,
135 contract_t, internal::packet_traits<RhsScalar>::size,
144 RhsScalar, Index, typename RhsMapper::SubMapper, Traits::nr, ColMajor>
146 typedef internal::gebp_kernel<LhsScalar, RhsScalar, Index, OutputMapper
[all...] |
| /external/eigen/Eigen/src/Core/ |
| SolveTriangular.h | 19 template<typename LhsScalar, typename RhsScalar, typename Index, int Side, int Mode, bool Conjugate, int StorageOrder>
53 typedef typename Rhs::Scalar RhsScalar;
56 typedef Map<Matrix<RhsScalar,Dynamic,1>, Aligned> MappedRhs;
65 ei_declare_aligned_stack_constructed_variable(RhsScalar,actualRhs,rhs.size(),
71 triangular_solve_vector<LhsScalar, RhsScalar, Index, Side, Mode, LhsProductTraits::NeedToConjugate,
|
| GeneralProduct.h | 202 typedef typename Rhs::Scalar RhsScalar;
226 ComplexByReal = (NumTraits<LhsScalar>::IsComplex) && (!NumTraits<RhsScalar>::IsComplex),
231 typedef const_blas_data_mapper<RhsScalar,Index,RowMajor> RhsMapper;
232 RhsScalar compatibleAlpha = get_factor<ResScalar,RhsScalar>::run(actualAlpha);
239 <Index,LhsScalar,LhsMapper,ColMajor,LhsBlasTraits::NeedToConjugate,RhsScalar,RhsMapper,RhsBlasTraits::NeedToConjugate>::run(
265 compatibleAlpha = RhsScalar(1);
272 <Index,LhsScalar,LhsMapper,ColMajor,LhsBlasTraits::NeedToConjugate,RhsScalar,RhsMapper,RhsBlasTraits::NeedToConjugate>::run(
296 typedef typename Rhs::Scalar RhsScalar;
317 gemv_static_vector_if<RhsScalar,ActualRhsTypeCleaned::SizeAtCompileTime,ActualRhsTypeCleaned::MaxSizeAtCompileTime,!DirectlyUseRhs (…)
[all...] |
| /external/eigen/Eigen/src/Core/products/ |
| SelfadjointMatrixVector.h | 181 typedef typename Rhs::Scalar RhsScalar;
198 internal::gemv_static_vector_if<RhsScalar,ActualRhsTypeCleaned::SizeAtCompileTime,ActualRhsTypeCleaned::MaxSizeAtCompileTime,!UseRhs> static_rhs;
203 ei_declare_aligned_stack_constructed_variable(RhsScalar,actualRhsPtr,rhs.size(),
204 UseRhs ? const_cast<RhsScalar*>(rhs.data()) : static_rhs.data());
|
| TriangularMatrixVector.h | 17 template<typename Index, int Mode, typename LhsScalar, bool ConjLhs, typename RhsScalar, bool ConjRhs, int StorageOrder, int Version=Specialized>
20 template<typename Index, int Mode, typename LhsScalar, bool ConjLhs, typename RhsScalar, bool ConjRhs, int Version>
21 struct triangular_matrix_vector_product<Index,Mode,LhsScalar,ConjLhs,RhsScalar,ConjRhs,ColMajor,Version>
23 typedef typename ScalarBinaryOpTraits<LhsScalar, RhsScalar>::ReturnType ResScalar;
30 const RhsScalar* _rhs, Index rhsIncr, ResScalar* _res, Index resIncr, const RhsScalar& alpha);
33 template<typename Index, int Mode, typename LhsScalar, bool ConjLhs, typename RhsScalar, bool ConjRhs, int Version>
34 EIGEN_DONT_INLINE void triangular_matrix_vector_product<Index,Mode,LhsScalar,ConjLhs,RhsScalar,ConjRhs,ColMajor,Version>
36 const RhsScalar* _rhs, Index rhsIncr, ResScalar* _res, Index resIncr, const RhsScalar& alpha
[all...] |
| GeneralMatrixMatrix.h | 23 typename RhsScalar, int RhsStorageOrder, bool ConjugateRhs>
24 struct general_matrix_matrix_product<Index,LhsScalar,LhsStorageOrder,ConjugateLhs,RhsScalar,RhsStorageOrder,ConjugateRhs,RowMajor>
26 typedef gebp_traits<RhsScalar,LhsScalar> Traits;
28 typedef typename ScalarBinaryOpTraits<LhsScalar, RhsScalar>::ReturnType ResScalar;
32 const RhsScalar* rhs, Index rhsStride,
35 level3_blocking<RhsScalar,LhsScalar>& blocking,
40 RhsScalar, RhsStorageOrder==RowMajor ? ColMajor : RowMajor, ConjugateRhs,
52 typename RhsScalar, int RhsStorageOrder, bool ConjugateRhs>
53 struct general_matrix_matrix_product<Index,LhsScalar,LhsStorageOrder,ConjugateLhs,RhsScalar,RhsStorageOrder,ConjugateRhs,ColMajor>
56 typedef gebp_traits<LhsScalar,RhsScalar> Traits
[all...] |
| GeneralBlockPanelKernel.h | 82 * Given a m x k times k x n matrix product of scalar types \c LhsScalar and \c RhsScalar,
92 template<typename LhsScalar, typename RhsScalar, int KcFactor, typename Index>
95 typedef gebp_traits<LhsScalar,RhsScalar> Traits;
108 kdiv = KcFactor * (Traits::mr * sizeof(LhsScalar) + Traits::nr * sizeof(RhsScalar)),
124 const Index n_cache = (l2-l1) / (nr * sizeof(RhsScalar) * k);
166 k_div = KcFactor * (Traits::mr * sizeof(LhsScalar) + Traits::nr * sizeof(RhsScalar)),
211 if(remaining_l1 >= Index(Traits::nr*sizeof(RhsScalar))*k)
214 max_nc = remaining_l1 / (k*sizeof(RhsScalar));
219 max_nc = (3*actual_l2)/(2*2*max_kc*sizeof(RhsScalar));
222 Index nc = numext::mini<Index>(actual_l2/(2*k*sizeof(RhsScalar)), max_nc) & (~(Traits::nr-1))
[all...] |
