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

Method FromFullShape

lingvo/core/gshard_utils.py:263–272  ·  view source on GitHub ↗

Creates tiled sharding spec with uneven padding computed from shape.

(cls, full_shape: Sequence[int],
                    split_dims_mapping: List[int], device_mesh: np.ndarray)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

GetVarShardingFunction · 0.80

Calls 1

TensorShardingSpecClass · 0.85

Tested by

no test coverage detected