Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.h348 dnn::AlgorithmDesc algorithm_desc; in ConvolveWithAlgorithm() local
394 const dnn::AlgorithmDesc &algorithm_desc, dnn::ConvolutionKind kind, in ConvolveRunnerFromDesc()
411 const dnn::AlgorithmDesc &algorithm_desc, dnn::ConvolutionKind kind, in FusedConvolveRunnerFromDesc()
H A Ddnn.cc133 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in ConvolveRunnerFromDesc()
159 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in FusedConvolveRunnerFromDesc()
H A Ddnn.h1330 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in PrepareForConvolution()
2626 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc2891 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2940 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2990 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
3964 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
4264 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
4817 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in ConvolveRunnerFromDesc()
5063 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in FusedConvolveRunnerFromDesc()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2911 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
3119 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
3187 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in ConvolveRunnerFromDesc()