Computes a zero state for each sub-step. Args: theta: A `.NestedMap` object containing weight values of this layer and its children layers. prepared_inputs: An output from PrepareExternalInputs. batch_size: The number of items in the batch that FProp will process.
(self, theta, prepared_inputs, batch_size)
source not stored for this graph (policy: none)