Searched defs:algorithm_desc (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 342 dnn::AlgorithmDesc algorithm_desc; in ConvolveWithAlgorithm() local
494 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardDataWithAlgorithm() local
526 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardFilterWithAlgorithm() local
|
| D | dnn.h | 1282 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in PrepareForConvolution()
2594 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2831 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2875 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2921 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
3871 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
3926 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2908 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2956 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
|