Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dplanner_helpers.py29 TensorWriteData,
56 ) -> TensorWriteData:
67 return TensorWriteData(
83 tensor_data=TensorWriteData(
110 tensor_data=TensorWriteData(
H A Dplanner.py45 class TensorWriteData: class
59 tensor_data: Optional[TensorWriteData] = None
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_shards_wrapper.py18 TensorWriteData,
259 tensor_data=TensorWriteData(