Searched refs:_ProxyTensor (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/fx/experimental/ |
| D | proxy_tensor.py | 178 def set_proxy_slot(obj: Tensor, tracer: _ProxyTracer, proxy: _ProxyTensor) -> None: argument
205 assert isinstance(proxy, _ProxyTensor)
252 ) -> _ProxyTensor:
261 ) -> Union[_ProxyTensor, U]:
270 transform: Callable[[_ProxyTensor], R], argument
581 set_proxy_slot(tensor, tracer, _ProxyTensor(proxy, constant))
680 class _ProxyTensor: class
707 def fetch_object_proxy(tracer: _ProxyTracer, t: Tensor) -> Union[_ProxyTensor, Tensor]:
820 if isinstance(t, _ProxyTensor)
831 t.constant if isinstance(t, _ProxyTensor) else t
[all …]
|
| /external/pytorch/torch/_dynamo/ |
| D | compiled_autograd.py | 447 assert isinstance(proxy_tensor, torch.fx.experimental.proxy_tensor._ProxyTensor)
|