Home
last modified time | relevance | path

Searched refs:kernel_spatial_extend (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DGraphBuilder.cpp220 … 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 DGraphBuilder.h158 …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,