vitis_quantize.VitisQuantizer.dump_model - 1.4.1 English

This function dumps the simulation results of the quantized model, including weights and activation results.

vitis_quantize.VitisQuantizer.dump_model(
    model,
    dataset=None,
    output_dir=’./dump_results’,
    dump_float=False,
    weights_only=False)

Arguments

model

A tf.keras.Model object, the quantized model to dump.

dataset

A tf.data.Dataset, keras.utils.Sequence or np.numpy object, the dataset used to dump, not needed if weights_only is set to True.

output_dir

A string object, the directory to save the dump results.

weights_only

A bool object, set to True to only dump the weights, set to False will also dump the activation results.