Searched defs:ConvolutionDescriptor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | Descriptors.h | 109 struct ConvolutionDescriptor struct
114 …int dim, int* pad, int* stride, int * upscale /* aka dilation */, int groups, bool deterministic) { in set()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.cc | 125 const dnn::ConvolutionDescriptor& /*convolution_descriptor*/, in GetConvolveRunners()
181 const dnn::ConvolutionDescriptor& /*convolution_descriptor*/, in GetMIOpenConvolveAlgorithms()
738 ConvolutionDescriptor::ConvolutionDescriptor(int ndims) { in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
746 ConvolutionDescriptor::ConvolutionDescriptor() in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
|
| H A D | stream.h | 69 class ConvolutionDescriptor; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/ |
| H A D | Descriptors.h | 196 struct TORCH_CUDA_CPP_API ConvolutionDescriptor struct
201 …e, int dim, int* pad, int* stride, int * upscale /* aka dilation */, int groups, bool allow_tf32) { in set()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 531 using ConvolutionDescriptor = typedef
|