def make_quant_avg_pool(bit_width,
kernel_size,
stride,
signed,
quant_type=QUANT_TYPE):
return qnn.QuantAvgPool2d(kernel_size=kernel_size,
quant_type=quant_type,
signed=signed,
stride=stride,
bit_width=bit_width)
def make_quant_avg_pool(bit_width,
kernel_size=POOL_KERNEL_SIZE,
stride=POOL_STRIDE,
signed=POOL_SIGNED):
'''Helper for AveragePooling layers'''
quant_type = get_quant_type(bit_width)
return qnn.QuantAvgPool2d(kernel_size=kernel_size,
quant_type=quant_type,
signed=signed,
stride=stride,
min_overall_bit_width=1,
max_overall_bit_width=bit_width)