Home
last modified time | relevance | path

Searched refs:reserve_space_allocator (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h88 ScratchAllocator* reserve_space_allocator,
106 ScratchAllocator* reserve_space_allocator,
124 ScratchAllocator* reserve_space_allocator,
229 bool is_training, ScratchAllocator* reserve_space_allocator,
243 bool is_training, ScratchAllocator* reserve_space_allocator,
600 bool is_training, ScratchAllocator* reserve_space_allocator,
654 ScratchAllocator* reserve_space_allocator,
Dcuda_dnn.cc1760 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl() argument
1788 SE_ASSIGN_OR_RETURN(reserve_space, reserve_space_allocator->AllocateBytes( in DoRnnForwardImpl()
2141 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
2164 reserve_space_allocator, workspace_allocator, output_profile_result), in DoRnnForward()
2182 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
2205 reserve_space_allocator, workspace_allocator, output_profile_result), in DoRnnForward()
2224 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
2247 reserve_space_allocator, workspace_allocator, output_profile_result), in DoRnnForward()
3458 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward() argument
3466 is_training, reserve_space_allocator, workspace_allocator), in DoBatchNormalizationForward()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h115 ScratchAllocator* reserve_space_allocator,
133 ScratchAllocator* reserve_space_allocator,
151 ScratchAllocator* reserve_space_allocator,
267 bool is_training, ScratchAllocator* reserve_space_allocator,
281 bool is_training, ScratchAllocator* reserve_space_allocator,
752 ScratchAllocator* reserve_space_allocator,
Drocm_dnn.cc2096 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl() argument
2144 reserve_space_allocator->AllocateBytes(reserve_space_size_in_bytes); in DoRnnForwardImpl()
2592 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
2617 reserve_space_allocator, workspace_allocator); in DoRnnForward()
2634 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
2659 reserve_space_allocator, workspace_allocator); in DoRnnForward()
2677 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
3532 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward() argument
3552 bool is_training, ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward() argument
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc796 ScratchAllocator* reserve_space_allocator, in DoForward() argument
835 &output_c_data, is_training, reserve_space_allocator, in DoForward()
1492 CudnnRnnAllocatorInOutput<T> reserve_space_allocator(context, 3); in ComputeAndReturnAlgorithm() local
1518 sequence_lengths, time_major, &reserve_space_allocator, in ComputeAndReturnAlgorithm()
1720 CudnnRnnAllocatorInTemp<T> reserve_space_allocator(context); in MaybeAutoTune() local
1725 &reserve_space_allocator, &workspace_allocator, in MaybeAutoTune()
1733 Tensor reserve_space = reserve_space_allocator.get_allocated_tensor(0); in MaybeAutoTune()
Dfused_batch_norm_op.cc923 reserve_space_allocator; in operator ()() local
927 reserve_space_allocator.reset( in operator ()()
963 is_training, reserve_space_allocator.get(), in operator ()()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1023 ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward() argument
1042 ScratchAllocator* reserve_space_allocator, in DoBatchNormalizationForward() argument
2200 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
2221 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
2242 ScratchAllocator* reserve_space_allocator, in DoRnnForward() argument
Dstream.h258 ScratchAllocator *reserve_space_allocator,
283 ScratchAllocator *reserve_space_allocator,
1794 ScratchAllocator *reserve_space_allocator,
1812 ScratchAllocator *reserve_space_allocator,
1830 ScratchAllocator *reserve_space_allocator,
Dstream.cc354 ScratchAllocator *reserve_space_allocator, in ThenBatchNormalizationForward() argument
363 is_training, reserve_space_allocator, workspace_allocator)); in ThenBatchNormalizationForward()
405 ScratchAllocator *reserve_space_allocator, in ThenBatchNormalizationForward() argument
414 is_training, reserve_space_allocator, workspace_allocator)); in ThenBatchNormalizationForward()
4553 ScratchAllocator *reserve_space_allocator, in ThenRnnForward() argument
4562 reserve_space_allocator, workspace_allocator, output_profile_result); in ThenRnnForward()
4586 ScratchAllocator *reserve_space_allocator, in ThenRnnForward() argument
4595 reserve_space_allocator, workspace_allocator, output_profile_result); in ThenRnnForward()
4620 ScratchAllocator *reserve_space_allocator, in ThenRnnForward() argument
4629 reserve_space_allocator, workspace_allocator, output_profile_result); in ThenRnnForward()