ConvCastFusionPass

融合模式

将Conv2D算子跟Cast算子融合为一个Conv2D算子。

融合成

融合成

使用约束

• 当Conv2D的输入数据类型为float16，Cast输出数据类型为float32时，该融合生效。
• 当Conv2D或cast节点为动态时，不融合。
• 当Conv2D输入节点中有StridedRead时，不融合。
• 当Conv2D输出节点大于1个时，不融合。

支持的型号

Atlas 200/300/500 推理产品

Atlas 推理系列产品

Atlas 200/500 A2推理产品

Atlas 训练系列产品