Home
last modified time | relevance | path

Searched refs:inline_inbuilt_nn_modules (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_decorators.py442 if torch._dynamo.config.inline_inbuilt_nn_modules:
475 if not torch._dynamo.config.inline_inbuilt_nn_modules:
485 if not torch._dynamo.config.inline_inbuilt_nn_modules:
H A Dtest_dynamic_shapes.py83 if not config.inline_inbuilt_nn_modules:
H A Dtest_higher_order_ops.py1985 if not torch._dynamo.config.inline_inbuilt_nn_modules:
2117 if not torch._dynamo.config.inline_inbuilt_nn_modules:
3730 @config.patch(inline_inbuilt_nn_modules=True)
3747 if torch._dynamo.config.inline_inbuilt_nn_modules:
3780 @config.patch(inline_inbuilt_nn_modules=True)
3810 if torch._dynamo.config.inline_inbuilt_nn_modules:
3844 @config.patch(inline_inbuilt_nn_modules=True)
3874 @config.patch(inline_inbuilt_nn_modules=False)
H A Dtest_structured_trace.py373 if not torch._dynamo.config.inline_inbuilt_nn_modules:
H A Dtest_exceptions.py247 @torch._dynamo.config.patch(inline_inbuilt_nn_modules=True)
H A Dtest_modules.py1972 if torch._dynamo.config.inline_inbuilt_nn_modules:
2080 if torch._dynamo.config.inline_inbuilt_nn_modules:
2700 if torch._dynamo.config.inline_inbuilt_nn_modules:
2706 if torch._dynamo.config.inline_inbuilt_nn_modules:
3012 with torch._dynamo.config.patch(inline_inbuilt_nn_modules=True):
H A Dtest_activation_checkpointing.py1004 @torch._dynamo.config.patch(inline_inbuilt_nn_modules=True)
1268 @torch._dynamo.config.patch(inline_inbuilt_nn_modules=True)
H A Dtest_hooks.py771 @torch._dynamo.config.patch(inline_inbuilt_nn_modules=True)
H A Dtest_repros.py1216 """11""" if torch._dynamo.config.inline_inbuilt_nn_modules else """5"""
1695 15 if torch._dynamo.config.inline_inbuilt_nn_modules else 12, cnt.op_count
5388 with torch._dynamo.config.patch(inline_inbuilt_nn_modules=True):
5630 @torch._dynamo.config.patch(inline_inbuilt_nn_modules=False)
H A Dtest_misc.py5426 if torch._dynamo.config.inline_inbuilt_nn_modules:
10110 @torch._dynamo.config.patch(inline_inbuilt_nn_modules=False)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dmutation_guard.py112 and config.inline_inbuilt_nn_modules
H A Dconfig.py378 inline_inbuilt_nn_modules = not is_fbcode() variable
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Duser_defined.py982 return torch._dynamo.config.inline_inbuilt_nn_modules and method in (
1141 torch._dynamo.config.inline_inbuilt_nn_modules
H A Dbuilder.py1377 if torch._dynamo.config.inline_inbuilt_nn_modules:
1468 config.inline_inbuilt_nn_modules
1481 not config.inline_inbuilt_nn_modules or is_parameter_freezing()
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_compile.py311 inline_inbuilt_nn_modules=True,
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcudagraph_trees.py794 torch._dynamo.config.inline_inbuilt_nn_modules
1972 if torch._dynamo.config.inline_inbuilt_nn_modules:
H A Dutils.py1733 torch._dynamo.config.inline_inbuilt_nn_modules
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_inductor_freezing.py344 @torch._dynamo.config.patch(inline_inbuilt_nn_modules=False)
H A Dtest_cpu_select_algorithm.py70 dynamo_config.patch(inline_inbuilt_nn_modules=True),
H A Dtest_cudagraph_trees.py2323 if torch._dynamo.config.inline_inbuilt_nn_modules:
H A Dtest_torchinductor.py11118 if torch._dynamo.config.inline_inbuilt_nn_modules
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_control_flow.py1834 if torch._dynamo.config.inline_inbuilt_nn_modules:
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py1719 condition=torch._dynamo.config.inline_inbuilt_nn_modules,