Lines Matching refs:kernel_spatial_extend
220 … Size2D kernel_spatial_extend, unsigned int depth, PadStrideInfo conv_info, in add_convolution_node() argument
228 ARM_COMPUTE_ERROR_ON((kernel_spatial_extend.width == 0) || (kernel_spatial_extend.height == 0)); in add_convolution_node()
238 …set(get_dimension_idx(input_data_layout, DataLayoutDimension::WIDTH), kernel_spatial_extend.width); in add_convolution_node()
239 …t(get_dimension_idx(input_data_layout, DataLayoutDimension::HEIGHT), kernel_spatial_extend.height); in add_convolution_node()
277 … Size2D kernel_spatial_extend, unsigned int depth, PadStrideInfo deconv_info, in add_deconvolution_node() argument
283 ARM_COMPUTE_ERROR_ON((kernel_spatial_extend.width == 0) || (kernel_spatial_extend.height == 0)); in add_deconvolution_node()
293 …set(get_dimension_idx(input_data_layout, DataLayoutDimension::WIDTH), kernel_spatial_extend.width); in add_deconvolution_node()
294 …t(get_dimension_idx(input_data_layout, DataLayoutDimension::HEIGHT), kernel_spatial_extend.height); in add_deconvolution_node()
332 …wise_convolution_node(Graph &g, NodeParams params, NodeIdxPair input, Size2D kernel_spatial_extend, in add_depthwise_convolution_node() argument
337 ARM_COMPUTE_ERROR_ON((kernel_spatial_extend.width == 0) || (kernel_spatial_extend.height == 0)); in add_depthwise_convolution_node()
347 …set(get_dimension_idx(input_data_layout, DataLayoutDimension::WIDTH), kernel_spatial_extend.width); in add_depthwise_convolution_node()
348 …t(get_dimension_idx(input_data_layout, DataLayoutDimension::HEIGHT), kernel_spatial_extend.height); in add_depthwise_convolution_node()