MatMulUnsqueezeSqueezeFusionPass
融合模式
MatMul/MatMulV2/BatchMatmul/BatchMatmulV2支持一维输入场景下,需要将一维输入插入Unsqueeze算子扩成二维,输出插入Squeeze算子去掉对应的扩维轴。
使用约束
无
父主题: 图融合规则说明
MatMul/MatMulV2/BatchMatmul/BatchMatmulV2支持一维输入场景下,需要将一维输入插入Unsqueeze算子扩成二维,输出插入Squeeze算子去掉对应的扩维轴。
无