Searched defs:ShapeEnv (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | _fake_tensor_utils.py | 41 def extract(self, shape_env: ShapeEnv) -> SymNode:
90 def extract(self, shape_env: ShapeEnv) -> PySymType:
142 def extract(self, shape_env: ShapeEnv) -> PySymType:
192 def extract(self, key: _DispatchCacheKey, shape_env: ShapeEnv) -> SymInt:
227 def __init__(self, shape_env: Optional[ShapeEnv] = None) -> None:
|
| H A D | fake_tensor.py | 329 shape_env: Optional[ShapeEnv] = None,
1098 shape_env: Optional[ShapeEnv] = None,
|
| H A D | meta_utils.py | 690 shape_env: Optional[ShapeEnv] = None,
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | aot_autograd.py | 438 shape_env: Optional[ShapeEnv],
519 shape_env: Optional[ShapeEnv],
532 shape_env: Optional[ShapeEnv],
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | runtime_assert.py | 15 ShapeEnv = Any variable
57 shape_env: ShapeEnv,
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | recording.py | 193 def assert_equal(old: Optional[ShapeEnv], new: ShapeEnv) -> ShapeEnv:
|
| H A D | validator.py | 685 def new_with_shape_env(shape_env: ShapeEnv, fake) -> Any:
699 …def check_shapeenv_fails(shape_env: ShapeEnv, tracked_fakes: Optional[List[Any]]) -> Optional[Vali…
|
| H A D | symbolic_shapes.py | 2310 class ShapeEnv: class
2620 def check_equal(self, other: "ShapeEnv") -> None:
|
| H A D | sym_node.py | 148 def with_shape_env(self, shape_env: "ShapeEnv") -> "SymNode":
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | sizevars.py | 42 shape_env: ShapeEnv,
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | dynamic_shapes.py | 380 shape_env: "ShapeEnv",
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_misc.py | 9297 def _replay_and_check(self, shape_env: ShapeEnv):
|