Searched defs:trace_joint (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | runtime_wrappers.py | 152 def __init__(self, info, runtime_metadata, trace_joint): argument
169 def __init__(self, info, runtime_metadata, trace_joint): argument
188 def __init__(self, info, runtime_metadata, trace_joint): argument
198 def __init__(self, info, runtime_metadata, trace_joint): argument
236 def make_output_handler(info, runtime_metadata, trace_joint): argument
|
| D | traced_function_transforms.py | 285 def create_functionalized_rng_ops_wrapper(func, args, trace_joint=True) -> Any: argument
|