Searched refs:NativeFunctionWithDifferentiabilityInfo (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_inplace_or_view_type.py | 13 NativeFunctionWithDifferentiabilityInfo,
450 fn: NativeFunctionWithDifferentiabilityInfo, var: str argument
526 def emit_inplace_or_view_body(fn: NativeFunctionWithDifferentiabilityInfo) -> list[str]: argument
586 fn: NativeFunctionWithDifferentiabilityInfo, argument
607 fn: NativeFunctionWithDifferentiabilityInfo, argument
621 def use_derived(fn: NativeFunctionWithDifferentiabilityInfo) -> bool: argument
628 fn: NativeFunctionWithDifferentiabilityInfo, argument
652 fns_with_infos: list[NativeFunctionWithDifferentiabilityInfo], argument
|
| H A D | gen_autograd.py | 33 NativeFunctionWithDifferentiabilityInfo,
74 NativeFunctionWithDifferentiabilityInfo
|
| H A D | gen_view_funcs.py | 35 from torchgen.api.autograd import NativeFunctionWithDifferentiabilityInfo
311 fns_with_infos: list[NativeFunctionWithDifferentiabilityInfo], argument
|
| H A D | context.py | 4 from torchgen.api.autograd import NativeFunctionWithDifferentiabilityInfo as NFWDI
|
| H A D | gen_variable_type.py | 41 NativeFunctionWithDifferentiabilityInfo,
851 fns_with_diff_infos: list[NativeFunctionWithDifferentiabilityInfo], argument
939 fn: NativeFunctionWithDifferentiabilityInfo, argument
1047 fn: NativeFunctionWithDifferentiabilityInfo, key: str = "Default" argument
|
| /aosp_15_r20/external/pytorch/torchgen/api/ |
| H A D | autograd.py | 256 class NativeFunctionWithDifferentiabilityInfo: class
263 def dispatch_strategy(fn: NativeFunctionWithDifferentiabilityInfo) -> str: argument
585 ) -> list[NativeFunctionWithDifferentiabilityInfo]:
658 result: list[NativeFunctionWithDifferentiabilityInfo] = []
676 NativeFunctionWithDifferentiabilityInfo(
819 NativeFunctionWithDifferentiabilityInfo(
836 fn: NativeFunctionWithDifferentiabilityInfo, key: str = "Default" argument
|