Searched defs:DifferentiabilityInfo (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/api/ |
| H A D | autograd.py | 99 class DifferentiabilityInfo: class
369 FunctionSchema, dict[str, DifferentiabilityInfo]
584 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]], argument
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_autograd_functions.py | 450 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]] argument
463 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]], argument
497 differentiability_infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]], argument
542 def process_function(info: DifferentiabilityInfo, template: CodeTemplate) -> str:
|
| H A D | load_derivatives.py | 68 infos: dict[FunctionSchema, dict[str, DifferentiabilityInfo]], argument
|