TensorFlow (1.15) 版本 - vai_p_tensorflow - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文

您必须在剪枝前先创建 TensorFlow 会话,此会话中包含计算图和已由 TensorFlow 初始化器、检查点、SavedModel 等完成初始化的变量。Vitis 优化器 TensorFlow 会就地执行计算图剪枝,并提供相应方法用于导出已冻结且已剪枝的计算图。

存储器中经剪枝的计算图是稀疏的,保留了原有的权重形状,同时移除特定的通道并将其设置为零。已冻结且已剪枝的计算图导出后即为大小更小的精简计算图,这意味着非必要的通道均已移除。