from torchgen.model import NativeFunctionsGroup # Follows dispatcher calling convention, but: # - Mutable arguments not allowed. Meta functions are always # written in functional form. Look at FunctionSchema.signature() # - No tensor returns; instead we return a TensorMeta describing # the tensor in question def name(g: NativeFunctionsGroup) -> str: # use the overload name from the functional version return str(g.functional.func.name).replace(".", "_")