def METHOD_NAME(name, batch_size):
    class PyAddSubNet(nn.Module):
        """
        Simple AddSub network in PyTorch. This network outputs the sum and
        subtraction of the inputs.
        """
        def __init__(self):
            super(PyAddSubNet, self).__init__()
        def forward(self, input0, input1):
            return torch.sub(input0, input1, alpha=-1), torch.sub(
                input0, input1, alpha=1
            )
    model = PyAddSubNet()
    model.eval()
    batch_size = 1
    example_inputs = torch.zeros([8, 4], dtype=torch.int64), torch.zeros(
        [8, 4], dtype=torch.int64
    )
    model_neuron = torch_neuron.trace(model, example_inputs, dynamic_batch_size=True)
    model_neuron.save("{}.pt".format(name))