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

Method FProp

lingvo/core/gpipe.py:296–321  ·  view source on GitHub ↗

Round-robin every children cells in cell_tpl among worker devices. Args: theta: A NestedMap object containing weights' values of this layer and its children layers. *args: Input args Returns: A list contains one tensor of [batch_size, feature_height, feature_width

(self, theta, *args)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

FPropMethod · 0.45
FPropMethod · 0.45
CellFnMethod · 0.45

Calls 2

_ToTupleFunction · 0.70

Tested by

no test coverage detected