vitis_quantize.VitisQuantizer.dump_model - 2.5 日本語

この関数は、量子化されたモデルのシミュレーション結果をダンプします。これには、重みとアクティベーションの結果が含まれます。

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

引数

model

tf.keras.Model オブジェクト。ダンプする量子化モデル。

dataset

tf.data.Dataset、keras.utils.Sequence、または np.numpy オブジェクト。ダンプに使用されるデータセット。weights_only が True に設定されている場合は不要です。

output_dir

string オブジェクト。ダンプ結果を保存するディレクトリ。

weights_only

bool オブジェクト。True に設定すると、重みのみをダンプします。False に設定すると、アクティベーション結果もダンプします。