Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1132 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationForward()
1152 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationForward()
1182 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationBackward()
1198 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationBackward()
1551 ScratchAllocator* workspace_allocator) { in DoPoolForward()
1562 ScratchAllocator* workspace_allocator) { in DoPoolForward()
1573 ScratchAllocator* workspace_allocator) { in DoPoolForward()
1587 ScratchAllocator* workspace_allocator) { in DoPoolBackward()
1600 ScratchAllocator* workspace_allocator) { in DoPoolBackward()
1613 ScratchAllocator* workspace_allocator) { in DoPoolBackward()
[all …]
Dstream.cc354 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationForward()
377 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationBackward()
405 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationForward()
429 ScratchAllocator *workspace_allocator) { in ThenBatchNormalizationBackward()
621 DeviceMemory<double> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
643 DeviceMemory<float> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
664 ScratchAllocator *workspace_allocator) { in ThenPoolForward()
684 DeviceMemory<int8> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
707 ScratchAllocator *workspace_allocator) { in ThenPoolBackward()
734 ScratchAllocator *workspace_allocator) { in ThenPoolBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1801 ScratchAllocator* workspace_allocator) { in CreateRnnWorkspace()
1822 ScratchAllocator* workspace_allocator) { in CreateBatchNormForwardWorkspace()
1845 ScratchAllocator* workspace_allocator) { in CreateBatchNormBackwardWorkspace()
1885 ScratchAllocator* workspace_allocator, in DoRnnForwardImpl()
2105 ScratchAllocator* workspace_allocator, in DoRnnBackwardImpl()
2426 ScratchAllocator* workspace_allocator, in DoRnnForward()
2469 ScratchAllocator* workspace_allocator, in DoRnnForward()
2513 ScratchAllocator* workspace_allocator, in DoRnnForward()
2540 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
2616 ScratchAllocator* workspace_allocator, in DoRnnBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2102 ScratchAllocator* workspace_allocator, in CreateRnnWorkspace()
2148 ScratchAllocator* workspace_allocator) { in DoRnnForwardImpl()
2270 ScratchAllocator* workspace_allocator) { in DoRnnBackwardImpl()
2645 ScratchAllocator* workspace_allocator, in DoRnnForward()
2688 ScratchAllocator* workspace_allocator, in DoRnnForward()
2732 ScratchAllocator* workspace_allocator, in DoRnnForward()
2738 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
2814 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2843 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
3528 ScratchAllocator* workspace_allocator) { in DoBatchNormalizationForward()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dctc_loss_op.cc354 DnnScratchAllocator workspace_allocator(1LL << 32, ctx); in Compute() local
Dcudnn_rnn_ops.cc799 ScratchAllocator* workspace_allocator, in DoForward()
882 ScratchAllocator* workspace_allocator, in DoBackward()
1536 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeAndReturnAlgorithm() local
1763 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in MaybeAutotune() local
1949 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeImpl() local
Dfused_batch_norm_op.cc946 workspace_allocator; in operator ()() local
1128 workspace_allocator; in operator ()() local