Searched defs:reserve_space_allocator (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 1174 ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
1194 ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
2129 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2151 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2173 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
|
| H A D | stream.cc | 339 bool is_training, ScratchAllocator *reserve_space_allocator, in ThenBatchNormalizationForward()
393 bool is_training, ScratchAllocator *reserve_space_allocator, in ThenBatchNormalizationForward()
2046 ScratchAllocator *reserve_space_allocator, in ThenRnnForward()
2081 ScratchAllocator *reserve_space_allocator, in ThenRnnForward()
2117 ScratchAllocator *reserve_space_allocator, in ThenRnnForward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 799 ScratchAllocator* reserve_space_allocator, in DoForward()
1534 CudnnRnnAllocatorInOutput<T> reserve_space_allocator(context, 3); in ComputeAndReturnAlgorithm() local
1764 CudnnRnnAllocatorInTemp<T> reserve_space_allocator(context); in MaybeAutotune() local
|
| H A D | fused_batch_norm_op.cc | 933 reserve_space_allocator; in operator ()() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1944 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl()
2485 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2528 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2572 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
5419 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
5443 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
5468 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForwardImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 2150 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl()
2647 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2690 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2734 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
3648 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
3668 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward()
|