MCPcopy Create free account
hub / github.com/CDTrans/CDTrans / _pad_to_largest_tensor

Function _pad_to_largest_tensor

datasets/sampler_ddp.py:38–62  ·  view source on GitHub ↗

Returns: list[int]: size of the tensor, on each rank Tensor: padded tensor that has the max size

(tensor, group)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

all_gatherFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected