Home
last modified time | relevance | path

Searched refs:range_constraints (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/pytorch/torch/export/
H A D_unlift.py37 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 Dexported_program.py631 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.py1118 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 Dunflatten.py214 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 Dadd_runtime_assertions_for_constraints_pass.py48 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 Dnon_strict_utils.py327 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__.py174 range_constraints = make_constraints(
184 range_constraints=range_constraints,
H A Dutils.py221 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.py109 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.py52 range_constraints=copy.deepcopy(real_exported_program.range_constraints),
/aosp_15_r20/external/executorch/exir/
H A Dlowered_backend_module.py89 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 Dserialize.py304 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 Dexport_serialize.py372 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 Dschema.py384 range_constraints: Dict[str, RangeConstraint]
/aosp_15_r20/external/executorch/backends/xnnpack/
H A Dxnnpack_preprocess.py123 range_constraints=edge_program.range_constraints,
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dserialize.py365 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 Dschema.py378 range_constraints: Dict[str, RangeConstraint]
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_serialize.py378 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 Dtest_export.py548 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 Dbackend_api.py407 range_constraints=copy.deepcopy(tagged_exported_program.range_constraints),
/aosp_15_r20/external/executorch/exir/capture/
H A D_capture.py116 range_constraints={},
349 range_constraints={},
/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_core.py776 exported_program.range_constraints
/aosp_15_r20/external/pytorch/docs/source/
H A Dexport.ir_spec.rst68 - ``range_constraints`` (``Dict[sympy.Symbol, RangeConstraint]``): For programs
H A Dexport.rst390 * ``exported_program.range_constraints`` describes the ranges of each symbol