MCPcopy Create free account
hub / github.com/tensorflow/lingvo / TensorShardingSpec

Class TensorShardingSpec

lingvo/core/gshard_utils.py:237–427  ·  view source on GitHub ↗

Represents a sharding annotation for GShard/XLA.

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 5

FromFullShapeMethod · 0.85
AddLeadingDimsMethod · 0.85
RemoveLeadingDimsMethod · 0.85
RemoveDimMethod · 0.85
ReplicatedSpecMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…