Searched refs:ConditionalAccumulatorBase (Results 1 – 13 of 13) sorted by relevance
69 ->template Delete<ConditionalAccumulatorBase>( in ~ConditionalAccumulatorBaseOp()75 typedef std::function<Status(ConditionalAccumulatorBase**)> Creator;95 ConditionalAccumulatorBase* accumulator; in SetAccumulatorHandle()98 ->template LookupOrCreate<ConditionalAccumulatorBase>( in SetAccumulatorHandle()127 ConditionalAccumulatorBase* accumulator; in ComputeAsync()138 ConditionalAccumulatorBase* accumulator,151 ConditionalAccumulatorBase* accumulator; in Compute()159 ConditionalAccumulatorBase* accumulator) = 0;175 ConditionalAccumulatorBase* accumulator) = 0;178 ConditionalAccumulatorBase* accumulator) override { in Compute()[all …]
20 ConditionalAccumulatorBase::ConditionalAccumulatorBase( in ConditionalAccumulatorBase() function in tensorflow::ConditionalAccumulatorBase27 Status ConditionalAccumulatorBase::MatchesNodeDef(const NodeDef& node_def) { in MatchesNodeDef()37 Status ConditionalAccumulatorBase::SetGlobalStep(int64 new_global_step) { in SetGlobalStep()59 void ConditionalAccumulatorBase::TryTakeGrad(int num_required, in TryTakeGrad()105 void ConditionalAccumulatorBase::Cancel( in Cancel()133 bool ConditionalAccumulatorBase::TryAttemptLocked( in TryAttemptLocked()163 void ConditionalAccumulatorBase::FlushUnlocked() { in FlushUnlocked()185 bool ConditionalAccumulatorBase::TakeGradLockedHelper(OpKernelContext* ctx, in TakeGradLockedHelper()
35 return [this](ConditionalAccumulatorBase** ret) { in GetCreator()73 ConditionalAccumulatorBase* accumulator) override { in CheckSignature()99 ConditionalAccumulatorBase* accumulator, in CheckSignature()
35 return [this](ConditionalAccumulatorBase** ret) { in GetCreator()74 ConditionalAccumulatorBase* accumulator) override { in CheckSignature()102 ConditionalAccumulatorBase* accumulator, in CheckSignature()
34 class TypedConditionalAccumulatorBase : public ConditionalAccumulatorBase {39 : ConditionalAccumulatorBase(dtype, shape, name) {} in TypedConditionalAccumulatorBase()
34 ConditionalAccumulatorBase* accumulator) override { in Compute()72 ConditionalAccumulatorBase* accumulator) override { in Compute()
47 class ConditionalAccumulatorBase : public ResourceBase {53 ConditionalAccumulatorBase(const DataType& dtype,
1 path: "tensorflow.ConditionalAccumulatorBase"3 is_instance: "<class \'tensorflow.python.ops.data_flow_ops.ConditionalAccumulatorBase\'>"
4 is_instance: "<class \'tensorflow.python.ops.data_flow_ops.ConditionalAccumulatorBase\'>"
28 name: "ConditionalAccumulatorBase"
75 * @{tf.ConditionalAccumulatorBase}
1161 class ConditionalAccumulatorBase(object): class1240 class ConditionalAccumulator(ConditionalAccumulatorBase):1320 class SparseConditionalAccumulator(ConditionalAccumulatorBase):