Searched refs:TryTakeGrad (Results 1 – 3 of 3) sorted by relevance
60 void TryTakeGrad(int num_required, OpKernelContext* ctx,
64 void ConditionalAccumulatorBase::TryTakeGrad(int num_required, in TryTakeGrad() function in tensorflow::ConditionalAccumulatorBase
233 accumulator->TryTakeGrad(num_required_tensor->scalar<int32>()(), ctx, in ComputeAsync()