| 123456789101112 |
- 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(".", "_")
|
