mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
add link to native_functions.yaml
@ -38,7 +38,7 @@ You can write this in a small standalone script that defines the Function and ru
|
||||
|
||||
## III) Write native composite function and OpInfo
|
||||
|
||||
Now that you know that the formula is correct, you can implement a brand new native function in “native_functions.yaml” as a CompositeImplicit function.
|
||||
Now that you know that the formula is correct, you can implement a brand new native function in [native_functions.yaml](../blob/master/aten/src/ATen/native/native_functions.yaml) as a CompositeImplicit function.
|
||||
You can then add an OpInfo entry (in [common_methods_invocations.py](../blob/master/torch/testing/_internal/common_methods_invocations.py)) to be able to test the new operator.
|
||||
|
||||
## IV) Write custom formula in derivatives.yaml
|
||||
|
Reference in New Issue
Block a user