Searched defs:pooling_size (Results 1 – 11 of 11) sorted by relevance
33 size_t pooling_size, in select_ukernel()67 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_argmax_pooling2d_nhwc_f32() local214 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_argmax_pooling2d_nhwc_f32() local
55 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_unpooling2d_nhwc_x32() local186 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_unpooling2d_nhwc_x32() local
67 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_q8() local239 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_average_pooling2d_nhwc_f32() local424 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_average_pooling2d_nhwc_q8() local552 const size_t pooling_size = pooling_height * pooling_width; in xnn_setup_average_pooling2d_nhwc_f32() local
67 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_max_pooling2d_nhwc_u8() local196 const uint32_t pooling_size = pooling_height * pooling_width; in xnn_create_max_pooling2d_nhwc_f32() local358 const size_t pooling_size = pooling_height * pooling_width; in setup_max_pooling2d() local
26 const size_t pooling_size = state.range(3); in max_pooling_u8() local99 const size_t pooling_size = state.range(3); in max_pooling_f32() local
26 const size_t pooling_size = state.range(3); in average_pooling_q8() local
136 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() argument143 inline UnpoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
128 inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() argument135 …inline ArgmaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
139 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() function146 inline MaxPoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width) { in pooling_size() function
139 inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_size) { in pooling_size() function146 …inline AveragePoolingOperatorTester& pooling_size(uint32_t pooling_height, uint32_t pooling_width)… in pooling_size() function
302 size_t pooling_size; member330 size_t pooling_size; member355 size_t pooling_size; member388 size_t pooling_size; member425 size_t pooling_size; member