OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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