Searched defs:PySymType (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 191 obj: PySymType, tracer: _ProxyTracer, proxy: _PySymProxyType
197 obj: Union[PySymType, _AnyScriptObjectType, Tensor],
304 obj: PySymType,
312 obj: PySymType,
321 obj: PySymType,
333 obj: Union[Tensor, _AnyScriptObjectType, PySymType], argument
516 proxy_callable: Callable[Concatenate[PySymType, _P], Proxy],
688 def inner(e: PySymType) -> Union[int, bool, float, Proxy]:
720 tracer: _ProxyTracer, t: PySymType
726 tracer: _ProxyTracer, t: Union[Tensor, _AnyScriptObjectType, PySymType] argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | _fake_tensor_utils.py | 87 def from_sym_type(value: PySymType) -> _DeconstructedSymType:
131 self, value: Union[PySymType, _DeconstructedSymType, _InputBackref]
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | types.py | 61 PySymType: TypeAlias = Union[SymInt, SymFloat, SymBool] variable
|