Remove references to FullGemm.
This commit is contained in:
		
							parent
							
								
									a42fdd08f3
								
							
						
					
					
						commit
						d2a90e2923
					
				|  | @ -347,7 +347,7 @@ void ONNXGemmOp::inferShapes() { | |||
|   getResult().setType(RankedTensorType::get(dims, lhsTy.getElementType())); | ||||
| } | ||||
| 
 | ||||
| // FullGemm
 | ||||
| // GemmNoBias
 | ||||
| 
 | ||||
| void ONNXGemmNoBiasOp::inferShapes() { | ||||
|   // Cannot infer shape if no shape exists.
 | ||||
|  |  | |||
|  | @ -30,7 +30,7 @@ def HasOneUse : Constraint<CPred<"$0.hasOneUse()">>; | |||
| // Pattern-Match and Rewrite | ||||
| //===----------------------------------------------------------------------===// | ||||
| 
 | ||||
| // onnx.add(onnx.matmul(%X, %Y), %Z) = onnx.FullGemm(%X, %Y, %Z) | ||||
| // onnx.add(onnx.matmul(%X, %Y), %Z) = onnx.Gemm(%X, %Y, %Z) | ||||
| def MulAddToGemmOptPattern : Pat<(ONNXAddOp (ONNXMatMulOp:$res $m1, $m2), $m3), | ||||
|                                  (ONNXGemmOp $m1, $m2, $m3), | ||||
| 				 [(HasOneUse $res)]>; | ||||
|  |  | |||
|  | @ -114,7 +114,7 @@ public: | |||
|         op->getName().getStringRef() != "onnx.Identity" && | ||||
|         op->getName().getStringRef() != "onnx.MatMul" && | ||||
|         op->getName().getStringRef() != "onnx.Gemm" && | ||||
|         op->getName().getStringRef() != "onnx.FullGemm" && | ||||
|         op->getName().getStringRef() != "onnx.GemmNoBias" && | ||||
|         op->getName().getStringRef() != "onnx.Reshape" && | ||||
|         op->getName().getStringRef() != "onnx.Transpose") | ||||
|       return false; | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue