| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dynamic_shapes.py | 164 def create_symbolic_tensor(name, arg, shape_env, source=None, dynamic_dims=None): argument
194 def create_symtype(cls, pytype, shape_env, val, duck=True, **kwargs): argument
208 def create_symint(shape_env, i: int, duck=True, **kwargs) -> SymInt: argument
212 def create_symbool(shape_env, b: bool) -> SymBool: argument
216 def create_symfloat(shape_env, f: float) -> SymFloat: argument
1076 def _do_test(self, fn, inp1, inp2, shape_env, is_unary_fn): argument
1080 def _do_test2(self, fn, inp1, inp2, shape_env, is_unary_fn): argument
|
| H A D | test_proxy_tensor.py | 932 def _get_free_symbols(shape_env): argument
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 882 t: MetaTensorDesc, source, shape_env, callback argument
954 base: torch.Tensor, t: MetaTensorDesc, source=source, shape_env=shape_env argument
1031 shape_env=shape_env, argument
1591 shape_env=None, argument
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/shape_inference/ |
| H A D | infer_shape.py | 90 def mksym(shape_env, value, source, dynamic_dim): argument
|
| /aosp_15_r20/external/pytorch/test/fx/ |
| H A D | test_shape_inference.py | 19 def mksym(shape_env, value, source, dynamic_dim) -> None: argument
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | symbolic_shapes.py | 282 def resolve_unbacked_bindings(shape_env, bindings): argument
290 def rebind_unbacked(shape_env, n: torch.fx.Node, result): argument
584 def compute_unbacked_bindings(shape_env, example_value, old_example_value=None, peek=False): argument
852 def _constrain_symbol_range(shape_env, s: sympy.Symbol, compiler_min: int, compiler_max: int): argument
|
| H A D | recording.py | 97 def run(self, shape_env=None) -> Any: argument
|
| H A D | validator.py | 669 def bisect(shape_env): argument
|
| H A D | sym_node.py | 71 shape_env, argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | graph.py | 301 shape_env=None, argument
|
| H A D | sizevars.py | 68 def __init__(self, shape_env=None) -> None: argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | output_graph.py | 589 def shape_env(self): member in OutputGraph
|
| H A D | guards.py | 2019 def _produce_guard_code(self, guard, code_list, shape_env=False): argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 3939 def create_symtype(cls, pytype, shape_env, val): argument
|