Searched defs:autograd_function (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | autograd_function.py | 36 def __call__(self, autograd_function, *args, **kwargs): argument
90 def custom_function_call_grad(interpreter, autograd_function, *operands): argument
97 def generate_single_level_function(interpreter, autograd_function): argument
261 def has_overriden_vmap_rule(autograd_function): argument
277 def custom_function_call_vmap(interpreter, autograd_function, *operands, **kwargs): argument
357 def custom_function_call_vmap_generate_rule(interpreter, autograd_function, *operands): argument
372 interpreter, autograd_function, generate_vmap_rule, *operands argument
377 def vmapify_autograd_function(autograd_function, in_dims, batch_size, randomness): argument
|