|
21 | 21 | RQIntegerDivTemplate, RQSiGELUTemplate, SliceTemplate, TransposeTemplate, iGELUTemplate, iLayernormTemplate, \ |
22 | 22 | iRMSNormTemplate, iSoftmaxTemplate |
23 | 23 | from Deeploy.Targets.Generic.TypeCheckers import AddChecker, BatchNormChecker, ConcatChecker, ConvChecker, \ |
24 | | - ConvTransposeChecker, DebugPrintChecker, DequantChecker, DivChecker, DummyChecker, GatherChecker, GELUChecker, \ |
25 | | - GEMMChecker, LayerNormChecker, MatMulChecker, MaxPoolChecker, MulChecker, PadChecker, QuantChecker, \ |
26 | | - ReduceMeanChecker, ReduceSumChecker, ReluChecker, RequantShiftChecker, ReshapeChecker, RQIntegerDivChecker, \ |
27 | | - SliceChecker, SoftmaxChecker, TransposeChecker |
| 24 | + DebugPrintChecker, DequantChecker, DivChecker, DummyChecker, GatherChecker, GELUChecker, GEMMChecker, \ |
| 25 | + LayerNormChecker, MatMulChecker, MaxPoolChecker, MulChecker, PadChecker, QuantChecker, ReduceMeanChecker, \ |
| 26 | + ReduceSumChecker, ReluChecker, RequantShiftChecker, ReshapeChecker, RQIntegerDivChecker, SliceChecker, \ |
| 27 | + SoftmaxChecker, TransposeChecker |
28 | 28 |
|
29 | 29 | BasicTransformer = CodeTransformation([ArgumentStructGeneration(), MemoryManagementGeneration(), FutureGeneration()]) |
30 | 30 |
|
|
296 | 296 |
|
297 | 297 | BasicConvTransposeBindings = [ |
298 | 298 | NodeBinding( |
299 | | - ConvTransposeChecker( |
| 299 | + ConvChecker( |
300 | 300 | [PointerClass(type), PointerClass(type), PointerClass(type)], # input, weight, bias |
301 | 301 | [PointerClass(type)]), |
302 | 302 | ConvTransposeTemplate.referenceTemplate, |
303 | 303 | BasicTransformer) for type in FloatDataTypes |
304 | 304 | ] + [ |
305 | 305 | NodeBinding( |
306 | | - ConvTransposeChecker( |
| 306 | + ConvChecker( |
307 | 307 | [PointerClass(type), PointerClass(type)], # input, weight |
308 | 308 | [PointerClass(type)]), |
309 | 309 | ConvTransposeTemplate.referenceTemplate, |
|
0 commit comments