hub / github.com/MTLab/MorphMLP / _serialize_to_tensor

Function _serialize_to_tensor

slowfast/utils/distributed.py:161–186 · view source on GitHub ↗

Seriialize the tensor to ByteTensor. Note that only `gloo` and `nccl` backend is supported. Args: data (data): data to be serialized. group (group): pytorch dist group. Returns: tensor (ByteTensor): tensor that serialized.

(data, group)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

all_gather_unalignedFunction · 0.85

Calls 1

get_rankFunction · 0.85

Tested by

no test coverage detected