| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 88 ScratchAllocator* workspace_allocator,
106 ScratchAllocator* workspace_allocator,
124 ScratchAllocator* workspace_allocator,
149 ScratchAllocator* workspace_allocator,
174 ScratchAllocator* workspace_allocator,
199 ScratchAllocator* workspace_allocator,
432 ScratchAllocator* workspace_allocator) override;
440 ScratchAllocator* workspace_allocator) override;
448 ScratchAllocator* workspace_allocator) override;
456 ScratchAllocator* workspace_allocator) override;
[all …]
|
| D | cuda_dnn.cc | 1489 ScratchAllocator* workspace_allocator) { in CreateRnnWorkspace() argument
1500 return workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in CreateRnnWorkspace()
1521 ScratchAllocator* workspace_allocator, in DoRnnForwardImpl() argument
1535 workspace_allocator)) in DoRnnForwardImpl()
1681 ScratchAllocator* workspace_allocator, in DoRnnBackwardImpl() argument
1695 workspace_allocator)); in DoRnnBackwardImpl()
1887 ScratchAllocator* workspace_allocator, in DoRnnForward() argument
1909 reserve_space_allocator, workspace_allocator, output_profile_result), in DoRnnForward()
1928 ScratchAllocator* workspace_allocator, in DoRnnForward() argument
1950 reserve_space_allocator, workspace_allocator, output_profile_result), in DoRnnForward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 79 ScratchAllocator* workspace_allocator,
97 ScratchAllocator* workspace_allocator,
115 ScratchAllocator* workspace_allocator,
140 ScratchAllocator* workspace_allocator,
165 ScratchAllocator* workspace_allocator,
190 ScratchAllocator* workspace_allocator,
423 ScratchAllocator* workspace_allocator = nullptr) override;
431 ScratchAllocator* workspace_allocator = nullptr) override;
439 ScratchAllocator* workspace_allocator = nullptr) override;
449 ScratchAllocator* workspace_allocator = nullptr) override;
[all …]
|
| D | rocm_dnn.cc | 1970 ScratchAllocator* workspace_allocator, in CreateRnnWorkspace() argument
1985 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in CreateRnnWorkspace()
2016 ScratchAllocator* workspace_allocator) { in DoRnnForwardImpl() argument
2041 workspace_allocator, &workspace)) { in DoRnnForwardImpl()
2138 ScratchAllocator* workspace_allocator) { in DoRnnBackwardImpl() argument
2162 workspace_allocator, &workspace)) { in DoRnnBackwardImpl()
2340 ScratchAllocator* workspace_allocator, in DoRnnForward() argument
2364 reserve_space_allocator, workspace_allocator); in DoRnnForward()
2382 ScratchAllocator* workspace_allocator, in DoRnnForward() argument
2406 reserve_space_allocator, workspace_allocator); in DoRnnForward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1576 ScratchAllocator* workspace_allocator) = 0;
1584 ScratchAllocator* workspace_allocator) {
1595 ScratchAllocator* workspace_allocator) {
1606 ScratchAllocator* workspace_allocator) {
1620 ScratchAllocator* workspace_allocator) {
1633 ScratchAllocator* workspace_allocator) {
1646 ScratchAllocator* workspace_allocator) {
1682 ScratchAllocator* workspace_allocator) {
2132 ScratchAllocator* workspace_allocator,
2153 ScratchAllocator* workspace_allocator,
[all …]
|
| D | stream.h | 522 ScratchAllocator *workspace_allocator = nullptr);
529 ScratchAllocator *workspace_allocator = nullptr);
536 ScratchAllocator *workspace_allocator = nullptr);
543 ScratchAllocator *workspace_allocator = nullptr);
552 ScratchAllocator *workspace_allocator = nullptr);
561 ScratchAllocator *workspace_allocator = nullptr);
570 ScratchAllocator *workspace_allocator = nullptr);
584 ScratchAllocator *workspace_allocator = nullptr);
1753 ScratchAllocator *workspace_allocator,
1771 ScratchAllocator *workspace_allocator,
[all …]
|
| D | stream.cc | 1225 DeviceMemory<double> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward() argument
1228 PARAM(workspace_allocator)); in ThenPoolForward()
1234 workspace_allocator)); in ThenPoolForward()
1250 DeviceMemory<float> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward() argument
1253 PARAM(workspace_allocator)); in ThenPoolForward()
1259 workspace_allocator)); in ThenPoolForward()
1273 ScratchAllocator *workspace_allocator) { in ThenPoolForward() argument
1276 PARAM(workspace_allocator)); in ThenPoolForward()
1282 workspace_allocator)); in ThenPoolForward()
1295 DeviceMemory<int8> *output_data, ScratchAllocator *workspace_allocator) { in ThenPoolForward() argument
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 739 ScratchAllocator* workspace_allocator, in DoForward() argument
777 workspace_allocator, output_profile_result) in DoForward()
801 ScratchAllocator* workspace_allocator, in DoBackward() argument
858 workspace_allocator, output_profile_result) in DoBackward()
1310 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeAndReturnAlgorithm() local
1334 &workspace_allocator, /*output_profile_result=*/nullptr); in ComputeAndReturnAlgorithm()
1534 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in MaybeAutoTune() local
1538 &reserve_space_allocator, &workspace_allocator, in MaybeAutoTune()
1552 ¶ms_backprop, nullptr, true, &workspace_allocator, in MaybeAutoTune()
1708 CudnnRnnAllocatorInTemp<uint8> workspace_allocator(context); in ComputeImpl() local
[all …]
|