Searched defs:NativeFunctionWithDifferentiabilityInfo (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_inplace_or_view_type.py | 450 fn: NativeFunctionWithDifferentiabilityInfo, var: str
526 def emit_inplace_or_view_body(fn: NativeFunctionWithDifferentiabilityInfo) -> list[str]:
586 fn: NativeFunctionWithDifferentiabilityInfo,
607 fn: NativeFunctionWithDifferentiabilityInfo,
621 def use_derived(fn: NativeFunctionWithDifferentiabilityInfo) -> bool:
628 fn: NativeFunctionWithDifferentiabilityInfo,
652 fns_with_infos: list[NativeFunctionWithDifferentiabilityInfo],
|
| H A D | gen_variable_type.py | 851 fns_with_diff_infos: list[NativeFunctionWithDifferentiabilityInfo],
939 fn: NativeFunctionWithDifferentiabilityInfo,
1047 fn: NativeFunctionWithDifferentiabilityInfo, key: str = "Default"
|
| H A D | gen_view_funcs.py | 311 fns_with_infos: list[NativeFunctionWithDifferentiabilityInfo],
|
| /aosp_15_r20/external/pytorch/torchgen/api/ |
| H A D | autograd.py | 256 class NativeFunctionWithDifferentiabilityInfo: class
263 def dispatch_strategy(fn: NativeFunctionWithDifferentiabilityInfo) -> str:
836 fn: NativeFunctionWithDifferentiabilityInfo, key: str = "Default"
|