Searched refs:PythonTensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/exir/ |
| H A D | tracer.py | 183 return t.proxy if isinstance(t, PythonTensor) else t
226 class PythonTensor(torch.Tensor): class
250 assert isinstance(r, PythonTensor)
320 if isinstance(out_arg_val, PythonTensor) and out_arg_val.is_immutable:
349 if isinstance(args[0], PythonTensor):
366 return PythonTensor(e, proxy)
371 if isinstance(e, PythonTensor):
428 parameter_proxy_cache[n] = PythonTensor(attr_val, proxy)
511 return PythonTensor(arg, placeholder, is_immutable=True)
|
| H A D | control_flow.py | 61 PythonTensor,
80 if not isinstance(x, PythonTensor):
92 return PythonTensor(unwrap_functional(tmp), proxy)
|
| H A D | wrap.py | 16 from executorch.exir.tracer import PythonTensor, unwrap_functional
28 assert isinstance(unwrapped, PythonTensor)
|