Searched refs:depthwise_args (Results 1 – 3 of 3) sorted by relevance
106 const DepthwiseArgs &depthwise_args; member110 : strategy(strat), depthwise_args(dwargs), output_stage(os)160 switch (args.depthwise_args.activation.type) in initialise()163 ws->activation_max = static_cast<T>(args.depthwise_args.activation.param1); in initialise()220 return sizeof(T) * args.depthwise_args.input_channels; in get_element_size()276 return sizeof(T) * args.depthwise_args.input_channels * args.depthwise_args.channel_multiplier; in sizeof_output_buffer()304 …const auto n_output_channels = args.depthwise_args.input_channels * args.depthwise_args.channel_mu… in initialise()350 … 0 : sizeof(int32_t) * args.depthwise_args.channel_multiplier * args.depthwise_args.input_channels; in sizeof_bias()357 … 0 : sizeof(int32_t) * args.depthwise_args.channel_multiplier * args.depthwise_args.input_channels; in sizeof_requant_muls()364 … 0 : sizeof(int32_t) * args.depthwise_args.channel_multiplier * args.depthwise_args.input_channels; in sizeof_requant_shifts()
294 …zeof(T *) * args.strategy->get_output_rows() * args.depthwise_args.kernel_rows * args.depthwise_ar… in sizeof_input_rows()315 const auto kernel_points = args.depthwise_args.kernel_rows * args.depthwise_args.kernel_cols; in sizeof_input_patch()
188 const auto kernel_points = args.depthwise_args.kernel_rows * args.depthwise_args.kernel_cols; in get_element_size()