MCPcopy Create free account
hub / github.com/InfiniTensor/InfiniTensor / shard_tensor

Function shard_tensor

examples/distributed/parallel.py:34–41  ·  view source on GitHub ↗
(tensor: TensorProto, dim: int)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

shard_gemmFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected