def METHOD_NAME(module, input, output, name):
    try:
        input_shapes = tuple(i.shape for i in input) if isinstance(input, tuple) else input.shape
        output_shapes = tuple(o.shape for o in output) if isinstance(output, tuple) else output.shape
        shapes[name] = (input_shapes, output_shapes)
    except:
        pass