in py/fpga_resnet.py [0:0]
def __init__(self, ext, context, program, queue,
inplanes, planes, stride=1, downsample=None):
self.ext = ext
self.conv1 = ext.Conv2d(context, program, queue,
inplanes, planes,
1, 1,
0, 0,
False, 0, 0)
self.relu1 = ext.ReLU(context, program, queue)
self.conv2 = ext.Conv2d(context, program, queue,
planes, planes,
3, stride,
1, 1,
False, 0, 0)
self.relu2 = ext.ReLU(context, program, queue)
self.conv3 = ext.Conv2d(context, program, queue,
planes, planes * self.expansion,
1, 1,
0, 0,
False, 0, 0)
self.relu3 = ext.ReLU(context, program, queue)
self.downsample = downsample
self.stride = stride
self.add = ext.Add(context, program, queue, 0, 0, 0)