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

Method MeshSplit

lingvo/core/batch_major_attention.py:8804–8807  ·  view source on GitHub ↗
(self, name, tensor_split_dims_mapping)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 11

FeedforwardMethod · 0.95
_DotAttenMethod · 0.45
_HeadsProjMethod · 0.45
_PostProjMethod · 0.45
ExtendStepMethod · 0.45
_DotAttenMethod · 0.45
_HeadsProjMethod · 0.45
FPropMethod · 0.45
FeedforwardMethod · 0.45
FeedforwardMethod · 0.45
_AttentionMethod · 0.45

Calls 2

SetMethod · 0.45
ParamsMethod · 0.45

Tested by

no test coverage detected