Home
last modified time | relevance | path

Searched defs:extract_tensor_metadata (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Daoti_eager.py94 def extract_tensor_metadata(dynamic: bool, input: torch.Tensor) -> Dict[str, Any]: function
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_tensor.py955 def extract_tensor_metadata(t: Tensor) -> TensorMetadata: function