Searched defs:fullgraph (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_compile.py | 151 def _reinplace_all_gather_with_optional_checks(self, fullgraph): argument
200 def _maybe_run_decide_global_ordering_of_comms_with_checks(self, fullgraph): argument
325 self, model_init_fn, input_creation_fn, backend, fullgraph argument
435 def _create_nested_fully_shard_factory_fns(self, fullgraph): argument
656 def _maybe_add_graph_break_to_sdpa(self, fullgraph): argument
657 def _sdpa_with_graph_break(orig_fn, fullgraph, *args, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_activation_checkpointing.py | 129 def _validate(self, fn, backend, *args, skip_check=False, fullgraph=True): argument
156 self, orig_fn, checkpointed_fn, *args, fullgraph=True argument
|
| H A D | test_higher_order_ops.py | 2934 def _compile_check(self, fn, inputs, fullgraph=True, graph_idx=0): argument
6272 def _validate(self, fn, backend, *args, skip_check=False, fullgraph=True): argument
|
| H A D | test_repros.py | 2969 def common_merge_criteria_processor_list(self, list_cls, fullgraph): argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_compiled_optimizers.py | 288 def compile_opt(opt_compiled, closure=None, fullgraph=True): argument
|
| H A D | test_compiled_autograd.py | 34 def make_compiler_fn(fullgraph=True, dynamic=True, backend="inductor"): argument
|