Searched refs:kernel_spatial_extend (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/src/graph/ |
H A D | GraphBuilder.cpp | 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() [all …]
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/ |
H A D | GraphBuilder.h | 158 …Size2D kernel_spatial_extend, unsigned int depth, PadStrideInfo conv_info, unsigned int num_groups… 177 … Size2D kernel_spatial_extend, unsigned int depth, PadStrideInfo deconv_info, 216 … Size2D kernel_spatial_extend, PadStrideInfo conv_info, int depth_multiplier = 1,
|