Searched defs:ConvolutionArgs (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/assembly/ |
| H A D | winograd.hpp | 37 struct ConvolutionArgs struct
39 unsigned int n_batches;
40 Shape2D input_shape;
41 unsigned int n_input_channels;
42 unsigned int pad_top, pad_left;
43 Shape2D output_shape;
44 unsigned int n_output_channels;
45 Shape2D kernel_shape;
46 arm_gemm::Activation activation;
48 ConvolutionArgs( in ConvolutionArgs() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | Conv_v7.cpp | 75 struct ConvolutionArgs { struct
76 cudnnHandle_t handle;
77 ConvolutionParams params;
78 TensorDescriptor idesc, odesc;
79 FilterDescriptor wdesc;
80 const Tensor &input, output, weight;
81 ConvolutionDescriptor cdesc;
83 ConvolutionArgs( in ConvolutionArgs() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | Conv_miopen.cpp | 224 struct ConvolutionArgs { struct
225 miopenHandle_t handle;
226 ConvolutionParams params;
227 TensorDescriptor idesc, odesc;
228 FilterDescriptor wdesc;
229 const Tensor& input, output, weight;
230 ConvolutionDescriptor cdesc;
232 …ConvolutionArgs(const Tensor& input, const Tensor& output, const Tensor& weight) : input(input), o… in ConvolutionArgs() argument
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/ |
| H A D | winograd_implementations.hpp | 59 inline bool constraints_met(const MethodConstraints &c, const CPUInfo *ci, const ConvolutionArgs &,… in constraints_met()
|
| H A D | output_transform.hpp | 59 virtual void initialise_thread_working_space(const ConvolutionArgs &, void *) const in initialise_thread_working_space() argument
|
| H A D | input_transform.hpp | 68 virtual void initialise_thread_working_space(const ConvolutionArgs &, void *) const in initialise_thread_working_space() argument
|