Home
last modified time | relevance | path

Searched defs:reserve_space_allocator (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1023 ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
1042 ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
2200 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2221 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2242 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
Dstream.cc354 ScratchAllocator *reserve_space_allocator, in ThenBatchNormalizationForward()
405 ScratchAllocator *reserve_space_allocator, in ThenBatchNormalizationForward()
4553 ScratchAllocator *reserve_space_allocator, in ThenRnnForward()
4586 ScratchAllocator *reserve_space_allocator, in ThenRnnForward()
4620 ScratchAllocator *reserve_space_allocator, in ThenRnnForward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc796 ScratchAllocator* reserve_space_allocator, in DoForward()
1492 CudnnRnnAllocatorInOutput<T> reserve_space_allocator(context, 3); in ComputeAndReturnAlgorithm() local
1720 CudnnRnnAllocatorInTemp<T> reserve_space_allocator(context); in MaybeAutoTune() local
Dfused_batch_norm_op.cc923 reserve_space_allocator; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1760 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl()
2141 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2182 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2224 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
3458 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
3481 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
3506 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2096 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl()
2592 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2634 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2677 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
3532 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
3552 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()