TopK

规则原文

规则定位：量化或非量化模式

本页是 ax quant spec 的逐算子展开。本页与统一规范冲突时，以统一规范为准。

规则

• 量化模式下，X 与每个已使用或作为 graph output 的 Values 输出共享完整量化参数。
• Values 完全未使用且不是 graph output 时不要求 Q/DQ。
• Indices 使用原生整数类型，不量化。
• 激活量化类型从 S8 对称或非对称、U8 非对称、S16 对称、U16 非对称中选择，粒度为 per-tensor。
• 非量化模式不拥有 Q/DQ。

本文档未增加限制的属性、shape 和输入形式，遵循模型声明 opset 下的 ONNX 语义。