def __init__()

in py/fpga_resnet.py [0:0]

• 25 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, ext, context, program, queue,
                 block, layers, num_classes=1000):
        self.inplanes = 64
        self.ext = ext

        self.conv1 = ext.Conv2d(context, program, queue,
                                3, 64,
                                7, 2,
                                3, 3, False, 0, 0)
        self.relu = ext.ReLU(context, program, queue)
        self.maxpool = ext.Pool2d(context, program, queue,
                                  3, 2, 1, 1, ext.PoolOp.Max, 0, 0)
        self.layer1 = self._make_layer(ext, context, program, queue,
                                       block, 64, layers[0])
        self.layer2 = self._make_layer(ext, context, program, queue,
                                       block, 128, layers[1], stride=2)
        self.layer3 = self._make_layer(ext, context, program, queue,
                                       block, 256, layers[2], stride=2)
        self.layer4 = self._make_layer(ext, context, program, queue,
                                       block, 512, layers[3], stride=2)
        self.avgpool = ext.Pool2d(context, program, queue,
                                  7, 1, 0, 0, ext.PoolOp.Avg, 0, 0)
        self.view = ext.View(context, program, queue,
                             [[0], [1, 2, 3]])
        self.fc = ext.Linear(context, program, queue,
                             512 * block.expansion, num_classes, True, 0, 0)