Searched defs:MetaTensorDesc (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 79 m1: Union[MetaTensorDesc, torch.Tensor],
442 class MetaTensorDesc: class
651 def get_tensor_memo(self, t: MetaTensorDesc):
654 def set_tensor_memo(self, t: MetaTensorDesc, v):
689 t: MetaTensorDesc,
742 t: MetaTensorDesc, src, symbolic_context=symbolic_context
776 inner_t: MetaTensorDesc, inner_src, symbolic_context=symbolic_context
793 t: MetaTensorDesc,
882 t: MetaTensorDesc, source, shape_env, callback
954 base: torch.Tensor, t: MetaTensorDesc, source=source, shape_env=shape_env
[all …]
|