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

Method FProp

lingvo/core/gpipe.py:421–605  ·  view source on GitHub ↗

Run multiple cells in different devices in a pipelining manner. Args: theta: A NestedMap object containing weights' values of this layer and its children layers. *args: Non-keyworded variable length argument list of input tensors. Returns: A list of output tensors

(self, theta, *args)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 10

_get_input_shapesMethod · 0.95
_get_state_dtypeMethod · 0.95
FilterMethod · 0.95
copyMethod · 0.80
_RecursiveMapMethod · 0.80
_ToTupleFunction · 0.70
FPropMethod · 0.45
TransformMethod · 0.45

Tested by

no test coverage detected