Home
last modified time | relevance | path

Searched refs:TensorPropertySource (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/test/
Dtest_dynamic_shapes.py1597 TensorPropertySource,
1612 TensorPropertySource(
1629 TensorPropertySource,
1632 src0 = TensorPropertySource(
1635 src2 = TensorPropertySource(
1638 src3 = TensorPropertySource(
1641 src4 = TensorPropertySource(
1645 src1 = TensorPropertySource(
1648 src7 = TensorPropertySource(
1652 src5 = TensorPropertySource(
[all …]
/external/pytorch/torch/fx/experimental/
Dsymbolic_shapes.py81 from torch._dynamo.source import TensorPropertySource
2989 from torch._dynamo.source import TensorPropertySource, TensorProperty
2997 TensorPropertySource(source, TensorProperty.SIZE, i),
3132 from torch._dynamo.source import TensorPropertySource, TensorProperty
3178 TensorPropertySource(source, TensorProperty.STRIDE, i),
3189 source=TensorPropertySource(source, TensorProperty.SIZE, i),
3199 … stride_expr, hint=ex_stride[i], source=TensorPropertySource(source, TensorProperty.STRIDE, i)))
3203 TensorPropertySource(source, TensorProperty.STORAGE_OFFSET),
3209 source=TensorPropertySource(source, TensorProperty.STORAGE_OFFSET))
3749 from torch._dynamo.source import TensorPropertySource, TensorProperty
[all …]
/external/pytorch/torch/_export/
Dnon_strict_utils.py15 TensorPropertySource,
106 src = TensorPropertySource(base=source, prop=TensorProperty.SIZE, idx=i)
/external/pytorch/torch/_dynamo/
Doutput_graph.py68 TensorPropertySource,
690 bind_symint(s, TensorPropertySource(src, TensorProperty.SIZE, i))
693 bind_symint(s, TensorPropertySource(src, TensorProperty.STRIDE, i))
696 TensorPropertySource(src, TensorProperty.STORAGE_OFFSET),
Dsource.py286 class TensorPropertySource(ChainedSource): class
Dguards.py55 TensorPropertySource,
1804 TensorPropertySource(source, TensorProperty.SIZE, dim)