Home
last modified time | relevance | path

Searched refs:compute_tensor_metadata (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/
H A D_content_store.py169 def compute_tensor_metadata(self, t: torch.Tensor, h=None): member in ContentStoreWriter
186 payload = self.compute_tensor_metadata(t, h=h)
/aosp_15_r20/external/pytorch/torch/_dynamo/repro/
H A Dafter_aot.py615 meta = writer.compute_tensor_metadata(val)
656 meta = writer.compute_tensor_metadata(r)