|
30 | 30 | RemoveEmptyConvBiasPass |
31 | 31 | from Deeploy.DeeployTypes import ConstantBuffer, DeploymentEngine, DeploymentPlatform, NodeMapper, NodeTemplate, \ |
32 | 32 | StructBuffer, TopologyOptimizer, TransientBuffer, VariableBuffer |
33 | | -from Deeploy.Targets.Generic.Bindings import BasicAddBindings, BasicConv1DBinding, BasicConv2DBindings, \ |
34 | | - BasicDebugPrintBindings, BasicDequantBindings, BasicDivBindings, BasicDWConv1DBinding, BasicDWConv2DBindings, \ |
35 | | - BasicGatherBindings, BasicGELUBindings, BasicGEMMBindings, BasicITAPartialSoftmaxBinding, BasicITASoftmaxBinding, \ |
36 | | - BasicLayerNormBindings, BasicMatMulBindings, BasicMaxPool2DBindings, BasicMulBindings, BasicPad1DBindings, \ |
37 | | - BasicPad2DBindings, BasicQuantBindings, BasicReduceMeanBindings, BasicReduceSumBindings, BasicReluBinding, \ |
38 | | - BasicReshapeBindings, BasicRQIntegerDivBinding, BasicRQSBindings, BasicRQSGELUBinding, BasicSliceBindings, \ |
39 | | - BasicSoftmaxBindings, BasicTransposeBindings, DummyBinding |
40 | | -from Deeploy.Targets.Generic.Layers import AddLayer, ConvLayer, DebugPrintLayer, DequantLayer, DivLayer, GatherLayer, \ |
41 | | - GELULayer, GEMMLayer, ITAMaxLayer, LayerNormLayer, MatMulLayer, MaxPoolLayer, MulLayer, PadLayer, QuantLayer, \ |
42 | | - ReduceMeanLayer, ReduceSumLayer, ReluLayer, RequantShiftLayer, ReshapeLayer, RQIntegerDivLayer, RQSiGELULayer, \ |
43 | | - SliceLayer, SoftmaxLayer, TransposeLayer |
44 | | -from Deeploy.Targets.Generic.Parsers import AddParser, DebugParser, DequantParser, DivParser, DummyParser, \ |
45 | | - FlattenParser, GatherParser, GELUParser, GenericConv1DParser, GenericConv2DParser, GenericDWConv1DParser, \ |
46 | | - GenericDWConv2DParser, GenericGEMMParser, GenericMaxPool2DParser, IntegerDivParser, ITAMaxParser, \ |
47 | | - ITAPartialMaxParser, LayerNormParser, MatMulParser, MulParser, Pad1DParser, Pad2DParser, QuantParser, \ |
48 | | - ReduceMeanParser, ReduceSumParser, ReluParser, RequantShiftParser, ReshapeParser, RQIntegerDivParser, \ |
| 33 | +from Deeploy.Targets.Generic.Bindings import BasicAddBindings, BasicConcatBindings, BasicConv1DBinding, \ |
| 34 | + BasicConv2DBindings, BasicDebugPrintBindings, BasicDequantBindings, BasicDivBindings, BasicDWConv1DBinding, \ |
| 35 | + BasicDWConv2DBindings, BasicGatherBindings, BasicGELUBindings, BasicGEMMBindings, BasicITAPartialSoftmaxBinding, \ |
| 36 | + BasicITASoftmaxBinding, BasicLayerNormBindings, BasicMatMulBindings, BasicMaxPool2DBindings, BasicMulBindings, \ |
| 37 | + BasicPad1DBindings, BasicPad2DBindings, BasicQuantBindings, BasicReduceMeanBindings, BasicReduceSumBindings, \ |
| 38 | + BasicReluBinding, BasicReshapeBindings, BasicRQIntegerDivBinding, BasicRQSBindings, BasicRQSGELUBinding, \ |
| 39 | + BasicSliceBindings, BasicSoftmaxBindings, BasicTransposeBindings, DummyBinding |
| 40 | +from Deeploy.Targets.Generic.Layers import AddLayer, ConcatLayer, ConvLayer, DebugPrintLayer, DequantLayer, DivLayer, \ |
| 41 | + GatherLayer, GELULayer, GEMMLayer, ITAMaxLayer, LayerNormLayer, MatMulLayer, MaxPoolLayer, MulLayer, PadLayer, \ |
| 42 | + QuantLayer, ReduceMeanLayer, ReduceSumLayer, ReluLayer, RequantShiftLayer, ReshapeLayer, RQIntegerDivLayer, \ |
| 43 | + RQSiGELULayer, SliceLayer, SoftmaxLayer, TransposeLayer |
| 44 | +from Deeploy.Targets.Generic.Parsers import AddParser, ConcatParser, DebugParser, DequantParser, DivParser, \ |
| 45 | + DummyParser, FlattenParser, GatherParser, GELUParser, GenericConv1DParser, GenericConv2DParser, \ |
| 46 | + GenericDWConv1DParser, GenericDWConv2DParser, GenericGEMMParser, GenericMaxPool2DParser, IntegerDivParser, \ |
| 47 | + ITAMaxParser, ITAPartialMaxParser, LayerNormParser, MatMulParser, MulParser, Pad1DParser, Pad2DParser, \ |
| 48 | + QuantParser, ReduceMeanParser, ReduceSumParser, ReluParser, RequantShiftParser, ReshapeParser, RQIntegerDivParser, \ |
49 | 49 | RQSiGELUParser, SliceParser, SoftmaxParser, TransposeParser, UnsqueezeParser, iLayerNormParser, iSoftmaxParser |
50 | 50 | from Deeploy.Targets.Generic.Templates import AllocateTemplate, FreeTemplate |
51 | 51 | from Deeploy.Targets.Generic.TopologyOptimizationPasses.Passes import DequantPatternPass, ExtractPaddingFromConvPass, \ |
|
55 | 55 | AddMapper = NodeMapper(AddParser(), BasicAddBindings) |
56 | 56 | Conv1DMapper = NodeMapper(GenericConv1DParser(), [BasicConv1DBinding]) |
57 | 57 | Conv2DMapper = NodeMapper(GenericConv2DParser(), BasicConv2DBindings) |
| 58 | +ConcatMapper = NodeMapper(ConcatParser(), BasicConcatBindings) |
58 | 59 | DebugMapper = NodeMapper(DebugParser(), BasicDebugPrintBindings) |
59 | 60 | DWConv1DMapper = NodeMapper(GenericDWConv1DParser(), [BasicDWConv1DBinding]) |
60 | 61 | DWConv2DMapper = NodeMapper(GenericDWConv2DParser(), BasicDWConv2DBindings) |
|
96 | 97 | GenericMapping = { |
97 | 98 | 'Add': AddLayer([AddMapper]), |
98 | 99 | 'Conv': ConvLayer([Conv2DMapper, DWConv2DMapper, Conv1DMapper, DWConv1DMapper]), |
| 100 | + 'Concat': ConcatLayer([ConcatMapper]), |
99 | 101 | 'DebugPrint': DebugPrintLayer([DebugMapper]), |
100 | 102 | 'Div': DivLayer([DivMapper]), |
101 | 103 | 'Flatten': ReshapeLayer([FlattenMapper]), |
|
0 commit comments