| /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
|