(self, sg_ir)
| 2094 | class MyLayer2(layer.Layer): |
| 2095 | |
| 2096 | def __init__(self, sg_ir): |
| 2097 | super(MyLayer2, self).__init__() |
| 2098 | self.sg_ir = sg_ir |
| 2099 | for node, operator in self.sg_ir.layers: |
| 2100 | self.__dict__[node.name] = operator |
| 2101 | self.conv2 = layer.Conv2d(1, 2) |
| 2102 | |
| 2103 | def forward(self, inputs): |
| 2104 | x = self.sg_ir.run(inputs, last_layers=-1)[0] |