tf.autodiff.ForwardAccumulator