Searched defs:num_required (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | conditional_accumulator_base.cc | 51 * A TakeGrad attempt is blocked until num_required > counter_, i.e.,
54 * num_required: Number of gradients that needs to be accumulated before the
59 void ConditionalAccumulatorBase::TryTakeGrad(int num_required, argument
62 if (num_required <= 0) {
64 "Argument num_required must be positive, but was ", num_required));
76 num_required, callback, ctx, cm, token,
|
Completed in 122 milliseconds