You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
482 B
13 lines
482 B
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(".", "_")
|