Init a SIGNA Model Args: onnx_model (ModelProto): a loaded onnx model
(self, onnx_model)
| 2196 | class SONNXModel(model.Model): |
| 2197 | |
| 2198 | def __init__(self, onnx_model): |
| 2199 | """ |
| 2200 | Init a SIGNA Model |
| 2201 | Args: |
| 2202 | onnx_model (ModelProto): a loaded onnx model |
| 2203 | """ |
| 2204 | super(SONNXModel, self).__init__() |
| 2205 | self.sg_ir = prepare(onnx_model) |
| 2206 | for node, operator in self.sg_ir.layers: |
| 2207 | self.__dict__[node.name] = operator |
| 2208 | self.sg_ir.is_graph = True |
| 2209 | |
| 2210 | def forward(self, *input, aux_output=(), **kwargs): |
| 2211 | """ |