def METHOD_NAME(self, x):
    x = self.conv1(x)
    x = self.bn1(x)
    x = tf.nn.relu(x)
    x = self.conv2(x)
    x = self.bn2(x)
    x = tf.nn.relu(x)
    x = self.flatten(x)
    x = self.fc(x)
    return x