HomeSort by relevance Sort by last modified time
    Searched refs:TryTakeGrad (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
conditional_accumulator_base.h 32 * The two main methods of this class are TryApplyGrad and TryTakeGrad.
39 * TryTakeGrad logs an attempt to read the average gradient. The attempt is
41 * or exceeds the number requested by TryTakeGrad.
60 void TryTakeGrad(int num_required, OpKernelContext* ctx,
141 // FlushUnlocked is called at the end of each TryApplyGrad and TryTakeGrad
conditional_accumulator_base.cc 59 void ConditionalAccumulatorBase::TryTakeGrad(int num_required,
conditional_accumulator_base_op.h 230 accumulator->TryTakeGrad(num_required_tensor->scalar<int32>()(), ctx,

Completed in 55 milliseconds