Home
last modified time | relevance | path

Searched refs:TensorBlockDescriptor (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorBlock.h188 class TensorBlockDescriptor {
245 friend class TensorBlockDescriptor;
258 static DestinationBuffer make(const TensorBlockDescriptor& desc, in make()
264 static DestinationBufferKind kind(const TensorBlockDescriptor& desc, in kind()
287 TensorBlockDescriptor(const IndexType offset, const Dimensions& dimensions, in TensorBlockDescriptor() function
293 TensorBlockDescriptor(const IndexType offset, const Dimensions& dimensions) in TensorBlockDescriptor() function
320 TensorBlockDescriptor& DropDestinationBuffer() { in DropDestinationBuffer()
331 TensorBlockDescriptor WithOffset(IndexType offset) const { in WithOffset()
332 return TensorBlockDescriptor(offset, m_dimensions, m_destination); in WithOffset()
348 typedef TensorBlockDescriptor<NumDims, IndexType> BlockDescriptor;
[all …]
H A DTensorEvaluator.h58 typedef internal::TensorBlockDescriptor<NumCoords, Index> TensorBlockDesc;
257 typedef internal::TensorBlockDescriptor<NumCoords, Index> TensorBlockDesc;
475 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
596 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
853 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorChipping.h154 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
157 typedef internal::TensorBlockDescriptor<NumInputDims, Index>
433 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorExecutor.h181 typedef internal::TensorBlockDescriptor<NumDims, StorageIndex>
362 typedef internal::TensorBlockDescriptor<NumDims, IndexType>
470 typedef internal::TensorBlockDescriptor<NumDims, IndexType> TensorBlockDesc;
H A DTensorMorphing.h136 typedef internal::TensorBlockDescriptor<NumOutputDims, Index> TensorBlockDesc;
268 typedef internal::TensorBlockDescriptor<TensorEvaluator::NumOutputDims, Index>
438 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
729 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorShuffling.h113 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
385 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorEvalTo.h123 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorForcedEval.h130 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorAssign.h120 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorGenerator.h106 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorReverse.h113 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorConversion.h341 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorPadding.h111 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
H A DTensorBroadcasting.h130 typedef internal::TensorBlockDescriptor<NumDims, Index> TensorBlockDesc;
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_access.cpp105 const internal::TensorBlockDescriptor<NumDims>& block, in UpdateCoeffSet()
231 typedef internal::TensorBlockDescriptor<5> TensorBlock; in test_uniform_block_shape()
358 typedef internal::TensorBlockDescriptor<5> TensorBlock; in test_skewed_inner_dim_block_shape()
H A Dcxx11_tensor_block_eval.cpp13 using Eigen::internal::TensorBlockDescriptor;
34 TensorBlockDescriptor<NumDims, Index> desc;
56 return {offsets, sizes, TensorBlockDescriptor<NumDims, Index>(offset, sizes)}; in RandomBlock()
104 return {offsets, dims, TensorBlockDescriptor<NumDims, Index>(0, dims)}; in FixedSizeBlock()