| /external/tensorflow/tensorflow/stream_executor/ |
| D | fft.h | 56 class ScratchAllocator; variable
110 ScratchAllocator *scratch_allocator) = 0;
115 ScratchAllocator *scratch_allocator) = 0;
120 bool in_place_fft, ScratchAllocator *scratch_allocator) = 0;
167 ScratchAllocator *scratch_allocator) = 0;
176 Stream *stream, Plan *plan, ScratchAllocator *scratch_allocator) = 0;
225 ScratchAllocator *scratch_allocator) override; \
228 bool in_place_fft, ScratchAllocator *scratch_allocator) override; \
231 fft::Type type, bool in_place_fft, ScratchAllocator *scratch_allocator) \
242 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) \
[all …]
|
| D | scratch_allocator.h | 40 class ScratchAllocator {
42 virtual ~ScratchAllocator();
65 class OneTimeScratchAllocator : public ScratchAllocator {
|
| D | dnn.h | 49 class ScratchAllocator; variable
1125 DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator,
1142 DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator,
1162 ScratchAllocator* scratch_allocator,
1180 DeviceMemory<int8>* output_data, ScratchAllocator* scratch_allocator,
1197 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc,
1576 ScratchAllocator* workspace_allocator) = 0;
1584 ScratchAllocator* workspace_allocator) {
1595 ScratchAllocator* workspace_allocator) {
1606 ScratchAllocator* workspace_allocator) {
[all …]
|
| D | scratch_allocator.cc | 23 ScratchAllocator::~ScratchAllocator() {} in ~ScratchAllocator()
|
| D | stream.h | 75 class ScratchAllocator; variable
311 DeviceMemory<double> *output, ScratchAllocator *scratch_allocator,
322 DeviceMemory<float> *output, ScratchAllocator *scratch_allocator,
333 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator,
347 DeviceMemory<double> *output, ScratchAllocator *scratch_allocator,
361 DeviceMemory<float> *output, ScratchAllocator *scratch_allocator,
376 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator,
390 ScratchAllocator *scratch_allocator,
412 ScratchAllocator *scratch_allocator,
424 ScratchAllocator *scratch_allocator,
[all …]
|
| D | stream.cc | 450 ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
488 ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
527 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
565 ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
600 ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
642 ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
684 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
829 ScratchAllocator *scratch_allocator, in ThenConvolveBackwardDataWithAlgorithm()
874 ScratchAllocator *scratch_allocator, in ThenConvolveBackwardDataWithAlgorithm()
919 ScratchAllocator *scratch_allocator, in ThenConvolveBackwardDataWithAlgorithm()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 55 ScratchAllocator* state_allocator) override;
87 ScratchAllocator* reserve_space_allocator,
88 ScratchAllocator* workspace_allocator,
105 ScratchAllocator* reserve_space_allocator,
106 ScratchAllocator* workspace_allocator,
123 ScratchAllocator* reserve_space_allocator,
124 ScratchAllocator* workspace_allocator,
149 ScratchAllocator* workspace_allocator,
174 ScratchAllocator* workspace_allocator,
199 ScratchAllocator* workspace_allocator,
[all …]
|
| D | cuda_fft.h | 72 int batch_count, ScratchAllocator* scratch_allocator);
77 ScratchAllocator* scratch_allocator);
80 ScratchAllocator *scratch_allocator);
|
| D | cuda_fft.cc | 80 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
236 ScratchAllocator *scratch_allocator) { in Initialize()
245 Stream *stream, ScratchAllocator *scratch_allocator) { in UpdateScratchAllocator()
310 ScratchAllocator *scratch_allocator) { in Create1dPlanWithScratchAllocator()
341 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create2dPlanWithScratchAllocator()
374 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create3dPlanWithScratchAllocator()
420 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
444 Stream *stream, fft::Plan *plan, ScratchAllocator *scratch_allocator) { in UpdatePlanWithScratchAllocator()
|
| D | cuda_dnn.cc | 935 ScratchAllocator* state_allocator) { in Create()
1038 ScratchAllocator* state_allocator) { in Create()
1489 ScratchAllocator* workspace_allocator) { in CreateRnnWorkspace()
1520 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl()
1521 ScratchAllocator* workspace_allocator, in DoRnnForwardImpl()
1681 ScratchAllocator* workspace_allocator, in DoRnnBackwardImpl()
1821 ScratchAllocator* state_allocator) { in createRnnDescriptor()
1886 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
1887 ScratchAllocator* workspace_allocator, in DoRnnForward()
1927 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
[all …]
|
| D | cuda_blas.h | 116 int batch_count, ScratchAllocator *scratch_allocator);
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 53 ScratchAllocator* state_allocator) override;
78 ScratchAllocator* reserve_space_allocator,
79 ScratchAllocator* workspace_allocator,
96 ScratchAllocator* reserve_space_allocator,
97 ScratchAllocator* workspace_allocator,
114 ScratchAllocator* reserve_space_allocator,
115 ScratchAllocator* workspace_allocator,
140 ScratchAllocator* workspace_allocator,
165 ScratchAllocator* workspace_allocator,
190 ScratchAllocator* workspace_allocator,
[all …]
|
| D | rocm_fft.h | 71 int batch_count, ScratchAllocator *scratch_allocator);
76 ScratchAllocator *scratch_allocator);
|
| D | rocm_fft.cc | 149 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
341 ScratchAllocator *scratch_allocator) { in Initialize()
390 ScratchAllocator *scratch_allocator) { in Create1dPlanWithScratchAllocator()
419 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create2dPlanWithScratchAllocator()
449 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create3dPlanWithScratchAllocator()
484 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
499 Stream *stream, fft::Plan *plan, ScratchAllocator *scratch_allocator) { in UpdatePlanWithScratchAllocator()
|
| D | rocm_dnn.cc | 1687 ScratchAllocator* state_allocator) in MIOpenRnnDescriptor()
1970 ScratchAllocator* workspace_allocator, in CreateRnnWorkspace()
2015 ScratchAllocator* reserve_space_allocator, in DoRnnForwardImpl()
2016 ScratchAllocator* workspace_allocator) { in DoRnnForwardImpl()
2138 ScratchAllocator* workspace_allocator) { in DoRnnBackwardImpl()
2280 ScratchAllocator* state_allocator) { in createRnnDescriptor()
2339 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2340 ScratchAllocator* workspace_allocator, in DoRnnForward()
2381 ScratchAllocator* reserve_space_allocator, in DoRnnForward()
2382 ScratchAllocator* workspace_allocator, in DoRnnForward()
[all …]
|
| D | rocm_blas.h | 109 int batch_count, ScratchAllocator *scratch_allocator);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | scratch_allocator.h | 30 class ScratchAllocator : public se::ScratchAllocator {
32 ScratchAllocator(int device_ordinal, DeviceMemoryAllocator* memory_allocator) in ScratchAllocator() function
|
| D | cusolver_rewriter.cc | 51 ScratchAllocator* allocator, in CreateCholesky()
134 ScratchAllocator* allocator, in RunOnInstruction()
187 ScratchAllocator scratch_allocator(device_ordinal, allocator); in RunOnComputation()
|
| D | scratch_allocator.cc | 21 StatusOr<se::DeviceMemory<uint8>> ScratchAllocator::AllocateBytes( in AllocateBytes()
|
| D | cudnn_conv_runner.h | 62 se::ScratchAllocator* scratch_allocator, se::Stream* stream,
|
| D | cudnn_conv_runner.cc | 80 class ScratchBufAllocator : public se::ScratchAllocator {
114 se::ScratchAllocator* scratch_allocator, in RunCudnnConvImpl()
407 se::ScratchAllocator* scratch_allocator, se::Stream* stream, in RunCudnnConv()
|
| D | fft_thunk.h | 38 class FftScratchAllocator : public se::ScratchAllocator {
|
| D | cudnn_conv_algorithm_picker.cc | 210 ScratchAllocator input_output_allocator(device_ordinal, allocator); in PickBestAlgorithm()
236 ScratchAllocator scratch_allocator(device_ordinal, allocator); in PickBestAlgorithm()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 123 using se::ScratchAllocator;
360 class CudnnRnnAllocatorInTemp : public ScratchAllocator {
407 class CudnnRnnAllocatorInOutput : public ScratchAllocator {
445 class CudnnRNNPersistentSpaceAllocator : public ScratchAllocator {
738 ScratchAllocator* reserve_space_allocator, in DoForward()
739 ScratchAllocator* workspace_allocator, in DoForward()
801 ScratchAllocator* workspace_allocator, in DoBackward()
980 ScratchAllocator* dropout_state_allocator, in CreateRnnDescriptor()
|
| D | conv_ops_gpu.h | 39 class DnnScratchAllocator : public se::ScratchAllocator {
|