Searched refs:pooling_params (Results 1 – 2 of 2) sorted by relevance
53 auto* pooling_params = layer_->mutable_pooling(); in Build() local56 pooling_params->set_type( in Build()58 pooling_params->set_globalpooling(true); in Build()64 pooling_params->mutable_stride()->Add(params->stride_height); in Build()65 pooling_params->mutable_stride()->Add(params->stride_width); in Build()66 pooling_params->mutable_kernelsize()->Add(params->filter_height); in Build()67 pooling_params->mutable_kernelsize()->Add(params->filter_width); in Build()70 pooling_params->mutable_same(); in Build()72 pooling_params->mutable_valid(); in Build()77 pooling_params->set_type( in Build()[all …]
908 const void* pooling_params = &average_pooling_op->params.f16_scaleminmax; in xnn_setup_average_pooling2d_nhwc_f16() local914 pooling_params = &average_pooling_op->params.f16_minmax; in xnn_setup_average_pooling2d_nhwc_f16()926 pooling_params, pooling_params_size, in xnn_setup_average_pooling2d_nhwc_f16()951 const void* pooling_params = &average_pooling_op->params.f32_scaleminmax; in xnn_setup_average_pooling2d_nhwc_f32() local957 pooling_params = &average_pooling_op->params.f32_minmax; in xnn_setup_average_pooling2d_nhwc_f32()969 pooling_params, pooling_params_size, in xnn_setup_average_pooling2d_nhwc_f32()