Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A Dstage.py1079 def _create_empty_tensors( function
1216 fwd_outputs = _create_empty_tensors(fwd_outputs, device)
1266 self.inputs = _create_empty_tensors(input_args, device)
1273 self.outputs = _create_empty_tensors(self.outputs, device)
1275 self.outputs = _create_empty_tensors(output_args, device)