/aosp_15_r20/external/pytorch/torch/export/ |
H A D | _unlift.py | 37 self.range_constraints, 241 def _create(cls, root, graph, range_constraints=None): argument 245 range_constraints=range_constraints, 250 def __init__(self, root, graph, range_constraints=None): argument 253 self.range_constraints = range_constraints or [] 258 range_constraints, argument 266 range_constraints=range_constraints, 358 new_gm, ep.range_constraints, ep.graph_signature
|
H A D | exported_program.py | 631 ep.range_constraints, 639 range_constraints=new_range_constraints, 669 range_constraints: "Dict[sympy.Symbol, Any]", 689 self._range_constraints: Dict[sympy.Symbol, ValueRanges] = range_constraints 762 def range_constraints(self): member in ExportedProgram 1089 range_constraints=_get_updated_range_constraints( 1091 self.range_constraints, 1112 input_placeholders, flat_args_with_path, self.range_constraints 1137 range_constraints=copy.deepcopy(self.range_constraints), 1171 range_constraints = copy.copy(old_range_constraints) [all …]
|
H A D | _trace.py | 1118 range_constraints = make_constraints( 1125 return range_constraints 1767 range_constraints = _get_range_constraints( 1791 range_constraints=range_constraints, 1902 range_constraints = _get_range_constraints( 1936 range_constraints=range_constraints,
|
H A D | unflatten.py | 214 self.range_constraints = export_module.range_constraints 502 self.input_placeholders, new_flat_args_with_path, self.range_constraints
|
/aosp_15_r20/external/pytorch/torch/_export/passes/ |
H A D | add_runtime_assertions_for_constraints_pass.py | 48 range_constraints: Dict[sympy.Symbol, ValueRanges], 51 self.range_constraints: Dict[sympy.Symbol, ValueRanges] = range_constraints 84 graph_module, self.range_constraints 119 constraint = self.range_constraints[symbol] 164 range_constraints: Dict[sympy.Symbol, ValueRanges], 216 if symint not in range_constraints:
|
/aosp_15_r20/external/pytorch/torch/_export/ |
H A D | non_strict_utils.py | 327 range_constraints = { 331 return range_constraints 362 range_constraints[d.node.expr] = shape_env.var_to_range[ 366 range_constraints[d.node.expr] = ValueRanges( 373 if symbol not in range_constraints: 378 range_constraints[symbol] = shape_env.var_to_range[symbol] 380 return range_constraints
|
H A D | __init__.py | 174 range_constraints = make_constraints( 184 range_constraints=range_constraints,
|
H A D | utils.py | 221 input_placeholders: List[torch.fx.Node], flat_args_with_path, range_constraints argument 313 if node_dim.node.expr in range_constraints: 315 range_constraints[node_dim.node.expr]
|
/aosp_15_r20/external/executorch/exir/program/ |
H A D | _program.py | 109 range_constraints = { 119 range_constraints[k] = v 120 return range_constraints 197 range_constraints=_get_updated_range_constraints(transformed_gm), 304 range_constraints, argument 314 range_constraints=range_constraints, 592 range_constraints=ep.exported_program.range_constraints, 632 range_constraints=new_ep.exported_program.range_constraints, 742 range_constraints=program.range_constraints,
|
H A D | _fake_program.py | 52 range_constraints=copy.deepcopy(real_exported_program.range_constraints),
|
/aosp_15_r20/external/executorch/exir/ |
H A D | lowered_backend_module.py | 89 range_constraints=copy.deepcopy( 90 self._original_exported_program.range_constraints 324 range_constraints=lowered_exported_program.range_constraints, 706 range_constraints=copy.deepcopy(owning_program.range_constraints),
|
/aosp_15_r20/external/executorch/exir/serde/ |
H A D | serialize.py | 304 exported_program.range_constraints 327 range_constraints=serialized_range_constraints, 596 for k, v in exported_program.range_constraints.items() 605 range_constraints = self.deserialize_range_constraints( 632 range_constraints=range_constraints,
|
H A D | export_serialize.py | 372 range_constraints: Dict[sympy.Symbol, ValueRanges] 379 for k, v in range_constraints.items() 1366 exported_program.range_constraints 1389 range_constraints=serialized_range_constraints, 2235 range_constraints = {} 2238 range_constraints[symbol] = v # type: ignore[arg-type] 2241 return range_constraints 2268 for k, v in exported_program.range_constraints.items() 2277 range_constraints = self.deserialize_range_constraints( 2293 range_constraints=range_constraints, [all …]
|
H A D | schema.py | 384 range_constraints: Dict[str, RangeConstraint]
|
/aosp_15_r20/external/executorch/backends/xnnpack/ |
H A D | xnnpack_preprocess.py | 123 range_constraints=edge_program.range_constraints,
|
/aosp_15_r20/external/pytorch/torch/_export/serde/ |
H A D | serialize.py | 365 range_constraints: Dict[sympy.Symbol, ValueRanges] 372 for k, v in range_constraints.items() 1375 exported_program.range_constraints 1391 range_constraints=serialized_range_constraints, 2283 range_constraints = {} 2286 range_constraints[symbol] = v # type: ignore[arg-type] 2289 return range_constraints 2312 for k, v in exported_program.range_constraints.items() 2324 range_constraints = self.deserialize_range_constraints( 2334 range_constraints=range_constraints, [all …]
|
H A D | schema.py | 378 range_constraints: Dict[str, RangeConstraint]
|
/aosp_15_r20/external/pytorch/test/export/ |
H A D | test_serialize.py | 378 range_constraints = list(ep.range_constraints.keys()) 379 assert len(range_constraints) == 1 380 symint = range_constraints[0] 386 ep.range_constraints[symint] = ValueRanges(lower=lower_range, upper=upper_range) 389 self.assertEqual(serialized.exported_program.range_constraints["s0"].min_val, 2) 390 self.assertEqual(serialized.exported_program.range_constraints["s0"].max_val, 3)
|
H A D | test_export.py | 548 vr.upper for vr in exported_program.range_constraints.values() 1804 str(ep_for_training.range_constraints), str(ep_for_real.range_constraints) 1941 self.assertEqual(len(ep.range_constraints), 3) 1942 for vr in ep.range_constraints.values(): 1976 vr = list(ep.range_constraints.values())[0] 1992 range_lower_bounds = sorted(vr.lower for vr in ep.range_constraints.values()) 1993 range_upper_bounds = sorted(vr.upper for vr in ep.range_constraints.values()) 2016 sym = next(iter(ep.range_constraints.keys())) 2041 sym = next(iter(ep.range_constraints.keys())) 2072 sym = next(iter(ep.range_constraints.keys()))
|
/aosp_15_r20/external/executorch/exir/backend/ |
H A D | backend_api.py | 407 range_constraints=copy.deepcopy(tagged_exported_program.range_constraints),
|
/aosp_15_r20/external/executorch/exir/capture/ |
H A D | _capture.py | 116 range_constraints={}, 349 range_constraints={},
|
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
H A D | _core.py | 776 exported_program.range_constraints
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | export.ir_spec.rst | 68 - ``range_constraints`` (``Dict[sympy.Symbol, RangeConstraint]``): For programs
|
H A D | export.rst | 390 * ``exported_program.range_constraints`` describes the ranges of each symbol
|