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

Function shard_tensor

examples/distributed/parallel_opt.py:21–44  ·  view source on GitHub ↗
(tensor: TensorProto, plc: Shard, groups: int = 1)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

shard_gemmFunction · 0.70
shard_binaryFunction · 0.70

Calls 2

is_shardMethod · 0.80
reshapeMethod · 0.80

Tested by

no test coverage detected