Home
last modified time | relevance | path

Searched refs:automatic_dynamic_shapes (Results 1 – 15 of 15) sorted by relevance

/external/pytorch/test/dynamo/
Dtest_config.py21 automatic_dynamic_shapes=False, assume_static_by_default=True
36 automatic_dynamic_shapes=True, assume_static_by_default=True
53 automatic_dynamic_shapes=True, assume_static_by_default=False
Dtest_cudagraphs.py48 verify_correctness=True, automatic_dynamic_shapes=True
Dtest_aot_autograd.py712 @torch._dynamo.config.patch(automatic_dynamic_shapes=False)
Dtest_misc.py6580 if torch._dynamo.config.automatic_dynamic_shapes:
7539 @torch._dynamo.config.patch(automatic_dynamic_shapes=False)
Dtest_repros.py1237 if torch._dynamo.config.automatic_dynamic_shapes:
/external/pytorch/benchmarks/dynamo/
Dtorchao_backend.py10 torch._dynamo.config.automatic_dynamic_shapes = False
/external/pytorch/torch/_dynamo/
Dconfig.py83 automatic_dynamic_shapes = True variable
Deval_frame.py292 automatic_dynamic_shapes=False, assume_static_by_default=True
1415 automatic_dynamic_shapes=False,
/external/pytorch/test/inductor/
Dtest_cuda_repro.py167 @dynamo_config.patch(automatic_dynamic_shapes=True)
195 @dynamo_config.patch(automatic_dynamic_shapes=True)
209 automatic_dynamic_shapes=True,
274 @dynamo_config.patch(automatic_dynamic_shapes=True)
288 @dynamo_config.patch(automatic_dynamic_shapes=True)
890 @dynamo_config.patch(automatic_dynamic_shapes=True)
Dtest_combo_kernels.py308 torch._dynamo.config.automatic_dynamic_shapes = False
Dtest_torchinductor.py6082 @dynamo_config.patch(automatic_dynamic_shapes=True)
6095 @dynamo_config.patch(automatic_dynamic_shapes=True)
7187 @dynamo_config.patch(automatic_dynamic_shapes=True)
7212 @dynamo_config.patch(automatic_dynamic_shapes=True)
8597 @dynamo_config.patch(automatic_dynamic_shapes=True)
Dtest_cudagraph_trees.py153 dynamo_config.patch(automatic_dynamic_shapes=True)
/external/pytorch/benchmarks/transformer/
Dscore_mod.py23 torch._dynamo.config.automatic_dynamic_shapes = False
/external/pytorch/torch/_dynamo/variables/
Dbuilder.py1789 config.automatic_dynamic_shapes and frame_state_entry.scalar is None
2572 automatic_dynamic_size = config.automatic_dynamic_shapes and (
2577 automatic_dynamic_stride = config.automatic_dynamic_shapes and (
/external/pytorch/test/functorch/
Dtest_control_flow.py3885 with torch._dynamo.config.patch(automatic_dynamic_shapes=True):