Searched refs:capture_dynamic_output_shape_ops (Results 1 – 19 of 19) sorted by relevance
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_torchinductor_dynamic_shapes.py | 217 @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 D | test_auto_functionalize.py | 340 capture_scalar_outputs=True, capture_dynamic_output_shape_ops=True
|
H A D | test_compiled_autograd.py | 2769 "test_setitem_mask": config.patch(capture_dynamic_output_shape_ops=True), 2771 capture_dynamic_output_shape_ops=True
|
H A D | test_triton_kernels.py | 1114 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
|
H A D | test_torchinductor.py | 4891 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | config.py | 229 capture_dynamic_output_shape_ops = ( variable
|
H A D | eval_frame.py | 1416 capture_dynamic_output_shape_ops=True,
|
H A D | output_graph.py | 299 allow_dynamic_output_shape_ops=config.capture_dynamic_output_shape_ops,
|
H A D | utils.py | 1805 if not torch._dynamo.config.capture_dynamic_output_shape_ops:
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_export.py | 2759 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 D | test_decorators.py | 578 torch._dynamo.config.capture_dynamic_output_shape_ops = True
|
H A D | test_misc.py | 8238 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 D | test_with_effects.py | 308 torch._dynamo.config.capture_dynamic_output_shape_ops = True
|
/aosp_15_r20/external/pytorch/test/distributed/ |
H A D | test_c10d_functional_native.py | 800 capture_dynamic_output_shape_ops=True,
|
H A D | test_inductor_collectives.py | 429 capture_dynamic_output_shape_ops=True,
|
H A D | test_dynamo_distributed.py | 426 @config.patch(optimize_ddp=True, capture_dynamic_output_shape_ops=True)
|
/aosp_15_r20/external/pytorch/test/export/ |
H A D | test_converter.py | 868 torch._dynamo.config.capture_dynamic_output_shape_ops = True
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
H A D | common.py | 3707 torch._dynamo.config.capture_dynamic_output_shape_ops = True
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_nestedtensor.py | 7343 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True) 7371 @torch._dynamo.config.patch(capture_dynamic_output_shape_ops=True)
|