Searched defs:workspace_allocator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1489 ScratchAllocator* workspace_allocator) { in CreateRnnWorkspace()
1521 ScratchAllocator* workspace_allocator, in DoRnnForwardImpl()
1681 ScratchAllocator* workspace_allocator, in DoRnnBackwardImpl()
1887 ScratchAllocator* workspace_allocator, in DoRnnForward()
1928 ScratchAllocator* workspace_allocator, in DoRnnForward()
1970 ScratchAllocator* workspace_allocator, in DoRnnForward()
2019 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2070 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2122 ScratchAllocator* workspace_allocator, in DoRnnBackward()
3733 DeviceMemory<double>* output_data, ScratchAllocator* workspace_allocator) { in DoPoolForward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1970 ScratchAllocator* workspace_allocator, in CreateRnnWorkspace()
2016 ScratchAllocator* workspace_allocator) { in DoRnnForwardImpl()
2138 ScratchAllocator* workspace_allocator) { in DoRnnBackwardImpl()
2340 ScratchAllocator* workspace_allocator, in DoRnnForward()
2382 ScratchAllocator* workspace_allocator, in DoRnnForward()
2425 ScratchAllocator* workspace_allocator, in DoRnnForward()
2454 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2505 ScratchAllocator* workspace_allocator, in DoRnnBackward()
2557 ScratchAllocator* workspace_allocator, in DoRnnBackward()
3480 DeviceMemory<double>* output_data, ScratchAllocator* workspace_allocator) { in DoPoolForward()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 739 ScratchAllocator* workspace_allocator, in DoForward()
801 ScratchAllocator* workspace_allocator, in DoBackward()
1310 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeAndReturnAlgorithm() local
1534 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in MaybeAutoTune() local
1708 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeImpl() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 1225 DeviceMemory<double> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
1250 DeviceMemory<float> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
1273 ScratchAllocator *workspace_allocator) { in ThenPoolForward()
1295 DeviceMemory<int8> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward()
1320 ScratchAllocator *workspace_allocator) { in ThenPoolBackward()
1350 ScratchAllocator *workspace_allocator) { in ThenPoolBackward()
1377 ScratchAllocator *workspace_allocator) { in ThenPoolBackward()
1420 ScratchAllocator *workspace_allocator) { in ThenNormalizeBackwardWithDimensions()
4861 ScratchAllocator *workspace_allocator, in ThenRnnForward()
4897 ScratchAllocator *workspace_allocator, in ThenRnnForward()
[all …]
|