Home
last modified time | relevance | path

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

/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()
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()
Dcuda_dnn.cc2215 ScratchAllocator* scratch_allocator) { in AllocateCudnnConvolutionForwardWorkspace()
2258 ScratchAllocator* scratch_allocator) { in AllocateCudnnConvolutionBackwardDataWorkspace()
2303 ScratchAllocator* scratch_allocator) { in AllocateCudnnConvolutionBackwardFilterWorkspace()
2346 const CudnnTensorDescriptor& output_nd, ScratchAllocator* scratch_allocator, in GetCudnnConvolutionForwardAlgorithm()
2395 const CudnnTensorDescriptor& output_nd, ScratchAllocator* scratch_allocator, in GetCudnnConvolutionBackwardDataAlgorithm()
2444 const CudnnTensorDescriptor& output_nd, ScratchAllocator* scratch_allocator, in GetCudnnConvolutionBackwardFilterAlgorithm()
2633 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2933 ScratchAllocator* scratch_allocator, in DoFusedConvolveImpl()
3353 DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3377 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.cc149 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()
Drocm_dnn.cc2599 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
3155 DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3172 DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3190 DeviceMemory<Eigen::half>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
3207 DeviceMemory<int8>* output_data, ScratchAllocator* scratch_allocator, in DoFusedConvolve()
Drocm_blas.cc1814 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal()
1910 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1931 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1948 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1967 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
1981 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc114 se::ScratchAllocator* scratch_allocator, in RunCudnnConvImpl()
399 ScratchBufAllocator scratch_allocator(scratch_buf); in RunCudnnConv() local
407 se::ScratchAllocator* scratch_allocator, se::Stream* stream, in RunCudnnConv()
Dcusolver_rewriter.cc187 ScratchAllocator scratch_allocator(device_ordinal, allocator); in RunOnComputation() local
Dfft_thunk.cc117 FftScratchAllocator scratch_allocator(buffer_allocations.device_ordinal(), in ExecuteOnStream() local
Dcudnn_conv_algorithm_picker.cc236 ScratchAllocator scratch_allocator(device_ordinal, allocator); in PickBestAlgorithm() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc454 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch() local
484 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch() local
Dconv_grad_ops_3d.cc1370 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute() local
1407 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in Compute() local
1776 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute() local
1814 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in Compute() local
Dconv_grad_filter_ops.cc850 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, in operator ()() local
883 DnnScratchAllocator scratch_allocator(ConvolveBackwardFilterScratchSize, ctx); in operator ()() local
Dconv_grad_input_ops.cc929 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, ctx); in operator ()() local
962 DnnScratchAllocator scratch_allocator(ConvolveBackwardDataScratchSize, in operator ()() local
Dconv_ops.cc864 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()() local
894 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in operator ()() local
Dfft_ops.cc416 CufftScratchAllocator scratch_allocator(CufftScratchSize, ctx); in DoFFT() local
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc718 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch() local
753 DnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx); in launch() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc450 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 …]