Lines Matching defs:accumulator
66 // If the accumulator object was not shared, delete it.
95 ConditionalAccumulatorBase* accumulator;
99 cinfo_.container(), cinfo_.name(), &accumulator, creator)));
100 core::ScopedUnref unref_me(accumulator);
102 // Verify that the shared accumulator is compatible
104 TF_RETURN_IF_ERROR(accumulator->MatchesNodeDef(def()));
127 ConditionalAccumulatorBase* accumulator;
129 ctx, GetResourceFromContext(ctx, "handle", &accumulator), callback);
130 ComputeAsync(ctx, accumulator, [callback, accumulator]() {
131 accumulator->Unref();
138 ConditionalAccumulatorBase* accumulator,
151 ConditionalAccumulatorBase* accumulator;
152 OP_REQUIRES_OK(ctx, GetResourceFromContext(ctx, "handle", &accumulator));
153 Compute(ctx, accumulator);
154 accumulator->Unref();
159 ConditionalAccumulatorBase* accumulator) = 0;
175 ConditionalAccumulatorBase* accumulator) = 0;
178 ConditionalAccumulatorBase* accumulator) override {
180 CheckSignature(ctx, accumulator);
192 accumulator->TryApplyGrad(local_step_tensor->scalar<int64>()(), ctx);
209 ConditionalAccumulatorBase* accumulator,
213 ConditionalAccumulatorBase* accumulator,
216 CheckSignature(ctx, accumulator, callback);
230 accumulator->TryTakeGrad(num_required_tensor->scalar<int32>()(), ctx,
