Searched defs:generate_vmap_rule (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ | ||
| H A D | autograd_function_db.py | 58 generate_vmap_rule = True variable in CubeGenVmap |
| /aosp_15_r20/external/pytorch/torch/autograd/ | ||
| H A D | function.py | 524 generate_vmap_rule = False variable in Function |
| /aosp_15_r20/external/pytorch/torch/_functorch/ | ||
| H A D | autograd_function.py | 372 interpreter, autograd_function, generate_vmap_rule, *operands argument |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_eager_transforms.py | 1368 generate_vmap_rule = True variable in TestAutogradFunctionVmapAPI.test_has_vmap_staticmethod_and_has_generate_vmap_rule.NumpyCube |