Home
last modified time | relevance | path

Searched defs:pooling_height (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/operators/
H A Dmax-pooling-nhwc.c42 uint32_t pooling_height, in create_max_pooling2d_nhwc()
272 const size_t pooling_height = max_pooling_op->kernel_height; in setup_max_pooling2d_nhwc() local
342 uint32_t pooling_height, in xnn_create_max_pooling2d_nhwc_s8()
382 uint32_t pooling_height, in xnn_create_max_pooling2d_nhwc_u8()
422 uint32_t pooling_height, in xnn_create_max_pooling2d_nhwc_f32()
476 uint32_t pooling_height, in xnn_create_max_pooling2d_nhwc_f16()
H A Daverage-pooling-nhwc.c42 uint32_t pooling_height, in xnn_create_average_pooling2d_nhwc_qu8()
250 uint32_t pooling_height, in xnn_create_average_pooling2d_nhwc_f16()
439 uint32_t pooling_height, in xnn_create_average_pooling2d_nhwc_f32()
719 const size_t pooling_height = average_pooling_op->kernel_height; in setup_average_pooling2d() local
H A Dunpooling-nhwc.c29 uint32_t pooling_height, in xnn_create_unpooling2d_nhwc_x32()
183 const size_t pooling_height = unpooling_op->kernel_height; in xnn_setup_unpooling2d_nhwc_x32() local
H A Dargmax-pooling-nhwc.c47 uint32_t pooling_height, in xnn_create_argmax_pooling2d_nhwc_f32()
196 const size_t pooling_height = argmax_pooling_op->kernel_height; in xnn_setup_argmax_pooling2d_nhwc_f32() local
/aosp_15_r20/external/XNNPACK/test/
H A Dunpooling-operator-tester.h143 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size()
151 inline UnpoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() function
157 inline uint32_t pooling_height() const { in pooling_height() function
H A Dargmax-pooling-operator-tester.h185 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size()
193 inline ArgmaxPoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() argument
199 inline uint32_t pooling_height() const { in pooling_height() function
H A Daverage-pooling-operator-tester.h195 …inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width)… in pooling_size()
203 inline AveragePoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() function
209 inline uint32_t pooling_height() const { in pooling_height() function
H A Dmax-pooling-operator-tester.h194 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size()
202 inline MaxPoolingOperatorTester& pooling_height(uint32_t pooling_height) { in pooling_height() argument
208 inline uint32_t pooling_height() const { in pooling_height() function
H A Dsubgraph-tester.h214 uint32_t pooling_height, uint32_t pooling_width, uint32_t stride_height, in AddAveragePooling2D()
312 uint32_t pooling_height, uint32_t pooling_width, uint32_t stride_height, in AddMaxPooling2D()
H A Daverage-pooling-2d.cc65 uint32_t pooling_height; member in AveragePoolingTestF32
H A Dargmax-pooling-2d.cc65 uint32_t pooling_height; member in ArgmaxPoolingTestF32
H A Dunpooling-2d.cc78 uint32_t pooling_height; member in Unpooling2DTestBase
H A Dmax-pooling-2d.cc90 uint32_t pooling_height; member in MaxPooling2DTestBase
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c272 const size_t pooling_height = op->kernel_height; in xnn_indirection_init_maxpool2d() local
837 const size_t pooling_height = op->kernel_height; in xnn_indirection_init_unpool2d() local
863 size_t pooling_height, in xnn_indirection_init_pavgpool2d_f16()
889 size_t pooling_height, in xnn_indirection_init_pavgpool2d_f32()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dmax-pooling.c40 uint32_t pooling_height, in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
H A Daverage-pooling.c35 uint32_t pooling_height, in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
H A Dindirection.c419 const size_t pooling_height = op->kernel_height; in pytorch_qnnp_indirection_init_maxpool2d() local
H A Doperator-run.c1336 const size_t pooling_height = op->kernel_height; in pytorch_qnnp_run_operator() local
1402 const size_t pooling_height = op->kernel_height; in pytorch_qnnp_run_operator() local
/aosp_15_r20/external/XNNPACK/src/subgraph/
H A Dunpooling-2d.c112 uint32_t pooling_height, in xnn_define_unpooling_2d()
H A Daverage-pooling-2d.c141 uint32_t pooling_height, in xnn_define_average_pooling_2d()
H A Dargmax-pooling-2d.c113 uint32_t pooling_height, in xnn_define_argmax_pooling_2d()
H A Dmax-pooling-2d.c221 uint32_t pooling_height, in xnn_define_max_pooling_2d()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dpool_2d_tester.h83 inline Pool2DTester& PoolingHeight(int32_t pooling_height) { in PoolingHeight()
H A Dquantized_pool_2d_tester.h84 inline QuantizedPool2DTester& PoolingHeight(int32_t pooling_height) { in PoolingHeight()
/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dsubgraph.h234 uint32_t pooling_height; member