tensorflow::ops::AccumulatorTakeGradient

tensorflow::ops::AccumulatorTakeGradient

#include <data_flow_ops.h>

Extracts the average gradient in the given ConditionalAccumulator.

Summary

The op blocks until sufficient (i.e., more than num_required) gradients have been accumulated. If the accumulator has already aggregated more than num_required gradients, it returns the average of the accumulated gradients. Also automatically increments the recorded global_step in the accumulator by 1, and resets the aggregate to 0.

Arguments:

  • scope: A Scope object
  • handle: The handle to an accumulator.
  • num_required: Number of gradients required before we return an aggregate.
  • dtype: The data type of accumu