Home
last modified time | relevance | path

Searched refs:TensorBlockScratch (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorExecutor.h184 TensorBlockScratch;
201 TensorBlockScratch scratch(device);
365 TensorBlockScratch;
379 TensorBlockScratch scratch(device);
391 TensorBlockScratch scratch(device);
472 TensorBlockScratch;
491 TensorBlockScratch scratch(ctx->device);
504 TensorBlockScratch scratch(ctx->device);
H A DTensorEvaluator.h59 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch; typedef
158 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
258 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
331 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
476 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
529 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
597 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
666 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
854 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
953 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
H A DTensorEvalTo.h124 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
173 TensorBlockDesc& desc, TensorBlockScratch& scratch) {
H A DTensorForcedEval.h131 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
207 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
H A DTensorAssign.h121 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
211 TensorBlockDesc& desc, TensorBlockScratch& scratch) {
H A DTensorMorphing.h137 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
200 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
439 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
616 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
730 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
H A DTensorGenerator.h107 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
183 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
H A DTensorBroadcasting.h131 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
629 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
851 TensorBlockScratch& scratch, ScalarNoConst* materialized_output,
1022 Index offset, TensorBlockScratch& scratch,
H A DTensorBlock.h729 template <typename TensorBlockScratch>
731 TensorBlockDesc& desc, TensorBlockScratch& scratch,
761 template <typename DataDimensions, typename TensorBlockScratch>
764 TensorBlockDesc& desc, TensorBlockScratch& scratch) {
H A DTensorReverse.h114 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
242 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
H A DTensorShuffling.h114 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
256 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
H A DTensorChipping.h155 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
290 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
H A DTensorConversion.h342 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
433 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
H A DTensorPadding.h112 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
239 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_eval.cpp129 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch; in VerifyBlockEvaluator() typedef
130 TensorBlockScratch scratch(d); in VerifyBlockEvaluator()