neural_compressor.tensorflow.keras.layers.conv2d

Initialize custom conv2d layers for Keras quantization.

Classes

QConv2D

The custom quantized Conv2D layer.

Functions

initialize_int8_conv2d(fp32_layer, q_config)

Initialize int8 conv2d.

Module Contents

class neural_compressor.tensorflow.keras.layers.conv2d.QConv2D(name, filters, kernel_size, strides=(1, 1), padding='valid', data_format=None, dilation_rate=(1, 1), groups=1, activation=None, use_bias=True, kernel_initializer='glorot_uniform', bias_initializer='zeros', kernel_regularizer=None, bias_regularizer=None, activity_regularizer=None, kernel_constraint=None, bias_constraint=None, act_min_value=None, act_max_value=None, weight_min_value=None, weight_max_value=None, granularity='per_tensor', quant_status='calib', quant_mode='SCALED', quant_T='s8', quant_round_mode='HALF_AWAY_FROM_ZERO', quant_narrow_range=False, quant_axis=None, **kwargs)

The custom quantized Conv2D layer.

neural_compressor.tensorflow.keras.layers.conv2d.initialize_int8_conv2d(fp32_layer, q_config)

Initialize int8 conv2d.