Home
last modified time | relevance | path

Searched defs:workspace_allocator (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1024 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationForward()
1043 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationForward()
1073 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationBackward()
1089 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationBackward()
1650 ScratchAllocator* workspace_allocator) { in DoPoolForward()
1661 ScratchAllocator* workspace_allocator) { in DoPoolForward()
1672 ScratchAllocator* workspace_allocator) { in DoPoolForward()
1686 ScratchAllocator* workspace_allocator) { in DoPoolBackward()
1699 ScratchAllocator* workspace_allocator) { in DoPoolBackward()
1712 ScratchAllocator* workspace_allocator) { in DoPoolBackward()
[all …]
Dstream.cc355 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationForward()
378 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationBackward()
406 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationForward()
430 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationBackward()
824 DeviceMemory<double> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
846 DeviceMemory<float> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
867 ScratchAllocator *workspace_allocator) { in ThenPoolForward()
887 DeviceMemory<int8> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
910 ScratchAllocator *workspace_allocator) { in ThenPoolBackward()
937 ScratchAllocator *workspace_allocator) { in ThenPoolBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1678 ScratchAllocator* workspace_allocator) { in CreateRnnWorkspace()
1699 ScratchAllocator* workspace_allocator) { in CreateBatchNormForwardWorkspace()
1722 ScratchAllocator* workspace_allocator) { in CreateBatchNormBackwardWorkspace()
1761 ScratchAllocator* workspace_allocator, in DoRnnForwardImpl()
1908 ScratchAllocator* workspace_allocator, in DoRnnBackwardImpl()
2142 ScratchAllocator* workspace_allocator, in DoRnnForward()
2183 ScratchAllocator* workspace_allocator, in DoRnnForward()
2225 ScratchAllocator* workspace_allocator, in DoRnnForward()
2274 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2325 ScratchAllocator* workspace_allocator, in DoRnnBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2051 ScratchAllocator* workspace_allocator, in CreateRnnWorkspace()
2097 ScratchAllocator* workspace_allocator) { in DoRnnForwardImpl()
2219 ScratchAllocator* workspace_allocator) { in DoRnnBackwardImpl()
2593 ScratchAllocator* workspace_allocator, in DoRnnForward()
2635 ScratchAllocator* workspace_allocator, in DoRnnForward()
2678 ScratchAllocator* workspace_allocator, in DoRnnForward()
2707 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2758 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2810 ScratchAllocator* workspace_allocator, in DoRnnBackward()
3533 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationForward()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc797 ScratchAllocator* workspace_allocator, in DoForward()
860 ScratchAllocator* workspace_allocator, in DoBackward()
1495 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeAndReturnAlgorithm() local
1721 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in MaybeAutoTune() local
1907 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeImpl() local
Dctc_loss_op.cc347 DnnScratchAllocator workspace_allocator(1LL << 32, ctx); in Compute() local
Dfused_batch_norm_op.cc925 workspace_allocator; in operator ()() local
1107 workspace_allocator; in operator ()() local