Hi, I have an exported torch script module that looks loosely like:
class Model(torch.nn.Module):
    def forward(self, tensor):
        // do forward pass

    @torch.jit.export
    def reinitialize_hidden_buffers(self, tensor):
        // reinitializes model parameters

When I load the torchscript module in Python, I can call reinitialize_hidden_buffers, but when the model is exported and loaded into objective c (for deployment on iOS),  I get
no member named 'reinitialize_hidden_buffers' in 'torch::jit::script::Module
Is there a way to call this correctly? Or should I just put control flow logic in the forward pass to do it.