Home
last modified time | relevance | path

Searched defs:algorithm_desc (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream.h337 dnn::AlgorithmDesc algorithm_desc; in ConvolveWithAlgorithm() local
448 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardDataWithAlgorithm() local
480 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardFilterWithAlgorithm() local
Ddnn.h1251 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in PrepareForConvolution()
1312 const dnn::AlgorithmDesc& algorithm_desc, in DoConvolve()
1415 const dnn::AlgorithmDesc& algorithm_desc, in DoConvolveBackwardData()
1462 const dnn::AlgorithmDesc& algorithm_desc, in DoConvolveBackwardFilter()
2686 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2539 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2583 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2629 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
2991 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
3046 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2851 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2946 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()