Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
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);
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,
DTensorEvalTo.h124 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
173 TensorBlockDesc& desc, TensorBlockScratch& scratch) {
DTensorAssign.h121 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
211 TensorBlockDesc& desc, TensorBlockScratch& scratch) {
DTensorForcedEval.h131 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
207 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
DTensorGenerator.h107 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
183 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
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;
DTensorBroadcasting.h131 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
629 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
851 TensorBlockScratch& scratch, ScalarNoConst* materialized_output,
1022 Index offset, TensorBlockScratch& scratch,
DTensorReverse.h114 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
242 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
DTensorShuffling.h114 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
256 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
DTensorBlock.h729 template <typename TensorBlockScratch>
731 TensorBlockDesc& desc, TensorBlockScratch& scratch,
761 template <typename DataDimensions, typename TensorBlockScratch>
764 TensorBlockDesc& desc, TensorBlockScratch& scratch) {
DTensorConversion.h342 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
433 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
DTensorChipping.h155 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
290 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
DTensorPadding.h112 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
239 block(TensorBlockDesc& desc, TensorBlockScratch& scratch,
/external/eigen/unsupported/test/
Dcxx11_tensor_block_eval.cpp129 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch; in VerifyBlockEvaluator() typedef
130 TensorBlockScratch scratch(d); in VerifyBlockEvaluator()