benchmarks/torch_ctl_benchmark/conv_model_benchmark.py/TorchModel/forward
class TorchModel:    def forward(self, x):
        x = self.conv(x)
        x = self.activation(x)
        x = self.max_pool(x)
        x = self.flatten(x)
        x = self.dense(x)
        x = self.softmax(x)
        return x