Home
last modified time | relevance | path

Searched refs:LocalSource (Results 1 – 17 of 17) sorted by relevance

/external/pytorch/test/dynamo/
Dtest_sources.py11 LocalSource,
21 x_src = LocalSource("x")
Dtest_misc.py42 from torch._dynamo.source import ConstantSource, GetItemSource, LocalSource
7605 s = GetItemSource(LocalSource("foo"), slice(None, -1, None))
7606 s2 = GetItemSource(LocalSource("foo"), slice(None, -1, None))
7607 s3 = GetItemSource(LocalSource("foo"), slice(None, -1, 2))
/external/pytorch/torch/fx/experimental/shape_inference/
Dinfer_shape.py6 from torch._dynamo.source import LocalSource
33 mksym(shape_env, v, LocalSource(k), DimDynamic.DYNAMIC)
/external/pytorch/test/fx/
Dtest_shape_inference.py9 from torch._dynamo.source import LocalSource
34 mksym(shape_env, v, LocalSource(k), DimDynamic.DYNAMIC)
/external/pytorch/test/
Dtest_dynamic_shapes.py1595 LocalSource,
1613 base=LocalSource(local_name="a"), prop=TensorProperty.SIZE, idx=0
1627 LocalSource,
1633 base=LocalSource(local_name="a"), prop=TensorProperty.SIZE, idx=0
1636 base=LocalSource(local_name="b"), prop=TensorProperty.SIZE, idx=0
1639 base=LocalSource(local_name="c"), prop=TensorProperty.SIZE, idx=0
1642 base=LocalSource(local_name="d"), prop=TensorProperty.SIZE, idx=0
1646 base=LocalSource(local_name="a"), prop=TensorProperty.SIZE, idx=2
1649 base=LocalSource(local_name="a"), prop=TensorProperty.SIZE, idx=3
1653 base=LocalSource(local_name="a"), prop=TensorProperty.SIZE, idx=1
[all …]
Dtest_proxy_tensor.py1815 from torch._dynamo.source import LocalSource
1817 …str(fx_g.shape_env.produce_guards(fx_placeholder_vals(fx_g), [LocalSource("a"), LocalSource("b")],…
1821 …str(fx_g.shape_env.produce_guards(fx_placeholder_vals(fx_g), [LocalSource("a"), LocalSource("b")],…
/external/pytorch/torch/_dynamo/
Dside_effects.py19 from .source import GlobalSource, LocalSource, Source
418 … var.mutable_local.source = LocalSource(cg.tempvars[var]) # type: ignore[attr-defined]
428 var.mutable_local.source = LocalSource(cg.tempvars[var])
Dcompiled_autograd.py12 from torch._dynamo.source import GetItemSource, LocalSource
83 return GetItemSource(LocalSource(name), idx)
Doutput_graph.py63 LocalSource,
884 if isinstance(source, LocalSource):
931 and isinstance(x.source.base, LocalSource)
945 and isinstance(arg.source, LocalSource)
972 x.source = LocalSource(visited[list_idx])
1060 if isinstance(v.source, LocalSource) and v.source.local_name == k:
Dsource.py98 class LocalSource(Source): class
717 if not isinstance(source, LocalSource):
Dguards.py92 LocalSource,
851 if istype(source, LocalSource):
1326 if isinstance(guard.originating_source, LocalSource):
Dsymbolic_convert.py73 LocalSource,
2718 source=LocalSource(k, cell_or_freevar=k in cells_and_freevars_set),
2789 if isinstance(source, LocalSource):
/external/pytorch/torch/_export/
Dnon_strict_utils.py13 LocalSource,
65 source: Source = LocalSource("args")
/external/pytorch/torch/_dynamo/variables/
Dbuilder.py76 LocalSource,
1216 self.source, LocalSource
1830 if self.tx.export and not isinstance(self.get_source(), LocalSource):
1906 if self.tx.export and not isinstance(self.get_source(), LocalSource):
1979 if self.tx.export and not isinstance(self.get_source(), LocalSource):
/external/python/bumble/bumble/
Davdtp.py1320 ) -> LocalSource:
1322 source = LocalSource(self, seid, codec_capabilities, packet_pump)
2177 class LocalSource(LocalStreamEndPoint): class
/external/pytorch/docs/source/
Dtorch.compiler_dynamo_deepdive.rst445 `LocalSource <https://github.com/pytorch/pytorch/blob/40dc0580a69565b06ec5263efe5d87cecc8200f7/torc…
448 which stores a ``LocalSource`` inside. On the other hand, ``a`` will not
479 ``[1]``) wrapping a ``LocalSource`` (``L['l']``).
/external/pytorch/torch/fx/experimental/
Dsymbolic_shapes.py3916 from torch._dynamo.source import LocalSource
3917 source = LocalSource(source)
4241 from torch._dynamo.source import LocalSource
4245 [LocalSource(a) for a in arg_names],