Home
last modified time | relevance | path

Searched defs:scratch_allocator (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc76 se::ScratchAllocator* scratch_allocator, in RunGpuConvForward()
96 se::ScratchAllocator* scratch_allocator, in RunGpuConvForwardActivation()
149 se::ScratchAllocator* scratch_allocator, in RunGpuConvInternalImpl()
195 se::ScratchAllocator* scratch_allocator, in RunGpuConvInternalImpl()
219 se::ScratchAllocator* scratch_allocator, in RunGpuConvImpl()
449 ScratchBufAllocator scratch_allocator(scratch_buf); in RunGpuConv() local
457 se::ScratchAllocator* scratch_allocator, se::Stream* stream, in RunGpuConv()
Dfft_thunk.cc118 FftScratchAllocator scratch_allocator(buffer_allocations.device_ordinal(), in ExecuteOnStream() local
Dgpu_conv_algorithm_picker.cc449 se::RedzoneAllocator scratch_allocator( in PickBestAlgorithmNoCacheCuda() local
699 ScratchAllocator scratch_allocator(device_ordinal, allocator); in PickBestAlgorithmNoCacheRocm() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc80 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()
309 ScratchAllocator *scratch_allocator) { in Create1dPlanWithScratchAllocator()
340 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create2dPlanWithScratchAllocator()
373 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create3dPlanWithScratchAllocator()
419 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
443 Stream *stream, fft::Plan *plan, ScratchAllocator *scratch_allocator) { in UpdatePlanWithScratchAllocator()
Dcuda_dnn.cc2467 ScratchAllocator* scratch_allocator) { in AllocateCudnnConvolutionForwardWorkspace()
2510 ScratchAllocator* scratch_allocator) { in AllocateCudnnConvolutionBackwardDataWorkspace()
2555 ScratchAllocator* scratch_allocator) { in AllocateCudnnConvolutionBackwardFilterWorkspace()
2602 const CudnnTensorDescriptor& output_nd, ScratchAllocator* scratch_allocator, in GetCudnnConvolutionForwardAlgorithm()
2655 const CudnnTensorDescriptor& output_nd, ScratchAllocator* scratch_allocator, in GetCudnnConvolutionBackwardDataAlgorithm()
2707 const CudnnTensorDescriptor& output_nd, ScratchAllocator* scratch_allocator, in GetCudnnConvolutionBackwardFilterAlgorithm()
2901 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
3224 ScratchAllocator* scratch_allocator, in DoFusedConvolveImpl()
3788 DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3812 DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
[all …]
Dcuda_blas.cc2150 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
2278 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2296 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2312 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2330 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2348 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.cc161 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize()
351 ScratchAllocator *scratch_allocator) { in Initialize()
400 ScratchAllocator *scratch_allocator) { in Create1dPlanWithScratchAllocator()
429 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create2dPlanWithScratchAllocator()
459 bool in_place_fft, ScratchAllocator *scratch_allocator) { in Create3dPlanWithScratchAllocator()
494 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator()
509 Stream *stream, fft::Plan *plan, ScratchAllocator *scratch_allocator) { in UpdatePlanWithScratchAllocator()
Drocm_blas.cc1820 int batch_count, uint64_t batch_stride, ScratchAllocator *scratch_allocator, in AllocateStridedBuffer()
1885 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
1977 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1998 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2015 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2034 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
2048 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
Drocm_dnn.cc2644 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
3444 DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3461 DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3479 DeviceMemory<Eigen::half>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3496 DeviceMemory<int8>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc255 DnnScratchAllocator scratch_allocator(PoolingScratchSize, context); in Compute() local
455 DnnScratchAllocator scratch_allocator(PoolingScratchSize, context); in Compute() local
Dconv_ops_3d.cc472 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch() local
567 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch() local
Dconv_grad_ops_3d.cc1392 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute() local
1502 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute() local
1874 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute() local
1952 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute() local
Dconv_grad_filter_ops.cc999 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in operator ()() local
1099 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, ctx); in operator ()() local
Dconv_ops.cc1014 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()() local
1113 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()() local
Dconv_grad_input_ops.cc1123 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()() local
1168 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in operator ()() local
Dfft_ops.cc468 CufftScratchAllocator scratch_allocator(CufftScratchSize, ctx); in DoFFT() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc470 ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
508 ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
547 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
585 ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
623 ScratchAllocator *scratch_allocator, in ThenFusedConvolveWithAlgorithm()
658 ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
700 ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
742 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
784 ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
826 ScratchAllocator *scratch_allocator, in ThenConvolveWithAlgorithm()
[all …]