Home
last modified time | relevance | path

Searched refs:capture_dynamic_output_shape_ops (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor_dynamic_shapes.py217 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
256 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
269 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
296 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
329 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
339 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
368 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
392 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
420 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
480 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
[all …]
H A Dtest_auto_functionalize.py340 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
H A Dtest_compiled_autograd.py2769 "test_setitem_mask": config.patch(capture_dynamic_output_shape_ops=True),
2771 capture_dynamic_output_shape_ops=True
H A Dtest_triton_kernels.py1114 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
H A Dtest_torchinductor.py4891 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dconfig.py229 capture_dynamic_output_shape_ops = ( variable
H A Deval_frame.py1416 capture_dynamic_output_shape_ops=True,
H A Doutput_graph.py299 allow_dynamic_output_shape_ops=config.capture_dynamic_output_shape_ops,
H A Dutils.py1805 if not torch._dynamo.config.capture_dynamic_output_shape_ops:
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_export.py2759 capture_dynamic_output_shape_ops=True,
2778 capture_dynamic_output_shape_ops=True,
3078 @config.patch(capture_dynamic_output_shape_ops=True)
3242 capture_dynamic_output_shape_ops=True,
3487 allow_dynamic_output_shape_ops=config.capture_dynamic_output_shape_ops,
3731 allow_dynamic_output_shape_ops=config.capture_dynamic_output_shape_ops,
H A Dtest_decorators.py578 torch._dynamo.config.capture_dynamic_output_shape_ops = True
H A Dtest_misc.py8238 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
8253 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
8298 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
10302 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
/aosp_15_r20/external/pytorch/test/higher_order_ops/
H A Dtest_with_effects.py308 torch._dynamo.config.capture_dynamic_output_shape_ops = True
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_functional_native.py800 capture_dynamic_output_shape_ops=True,
H A Dtest_inductor_collectives.py429 capture_dynamic_output_shape_ops=True,
H A Dtest_dynamo_distributed.py426 @config.patch(optimize_ddp=True, capture_dynamic_output_shape_ops=True)
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_converter.py868 torch._dynamo.config.capture_dynamic_output_shape_ops = True
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py3707 torch._dynamo.config.capture_dynamic_output_shape_ops = True
/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py7343 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
7371 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)