Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A Dstage.py1144 def _extract_metadata_from_tensor(tensor: torch.Tensor) -> List[torch.Size]: function
1207 metadata = _extract_metadata_from_tensor(input_shape_metadata_tensor)
1429 _extract_metadata_from_tensor(tensor) for tensor in tensor_list
1441 _extract_metadata_from_tensor(tensor) for tensor in tensor_list