Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST Register now

tensorflow:: ops:: SparseAccumulatorTakeGradient

#include <data_flow_ops.h>

Extracts the average sparse gradient in a SparseConditionalAccumulator .

Summary

The op will 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 will return its average of the accumulated gradients. Also automatically increments the recorded global_step in the accumulator by 1, and resets the aggregate to 0.

Args:

  • scope: A Scope object
  • handle: The handle to a SparseConditionalAccumulator .
  • num_required: Number of gradients required before we return an aggregate.
  • dtype: The data type of accumulated gradients. Needs to correspond to the type of the accumulator.

Returns:

  • Output indices: Indices of the average of the accumulated sparse gradients.
  • Output values: Values of the average of the accumulated sparse gradients.
  • Output shape: Shape of the average of the accumulated sparse gradients.

Constructors and Destructors

SparseAccumulatorTakeGradient (const :: tensorflow::Scope & scope, :: tensorflow::Input handle, :: tensorflow::Input num_required, DataType dtype)

Public attributes

indices
operation
shape
values

Public attributes

indices

::tensorflow::Output indices

operation

Operation operation

shape

::tensorflow::Output shape

values

::tensorflow::Output values

Public functions

SparseAccumulatorTakeGradient

 SparseAccumulatorTakeGradient(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_required,
  DataType dtype
)