MatmulCastFusionPass

Description

Fuses the MatMul/MatMulV2/BatchMatMul/BatchMatMulV2 and cast operators into the MatMul/MatMulV2/BatchMatMul/BatchMatMulV2 operator.

Before: After:

Constraints

This fusion takes effect when the input data type of MatMul is float16 and the output data type of cast is float32.

Applicable Products

Atlas inference products

Atlas training products