Searched defs:ConvolutionArgs (Results 1 – 4 of 4) sorted by relevance
37 struct ConvolutionArgs struct39 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() function
59 inline bool constraints_met(const MethodConstraints &c, const CPUInfo *ci, const ConvolutionArgs &,… in constraints_met()
59 virtual void initialise_thread_working_space(const ConvolutionArgs &, void *) const in initialise_thread_working_space() argument
68 virtual void initialise_thread_working_space(const ConvolutionArgs &, void *) const in initialise_thread_working_space() argument