Searched refs:ConditionalAccumulatorBase (Results 1 – 14 of 14) sorted by relevance
71 ->template Delete<ConditionalAccumulatorBase>( in ~ConditionalAccumulatorBaseOp()77 typedef std::function<Status(ConditionalAccumulatorBase**)> Creator;98 ConditionalAccumulatorBase* accumulator; in SetAccumulatorHandle()101 ->template LookupOrCreate<ConditionalAccumulatorBase>( in SetAccumulatorHandle()130 ConditionalAccumulatorBase* accumulator; in ComputeAsync()141 ConditionalAccumulatorBase* accumulator,154 ConditionalAccumulatorBase* accumulator; in Compute()162 ConditionalAccumulatorBase* accumulator) = 0;178 ConditionalAccumulatorBase* accumulator) = 0;181 ConditionalAccumulatorBase* accumulator) override { in Compute()[all …]
21 ConditionalAccumulatorBase::ConditionalAccumulatorBase( in ConditionalAccumulatorBase() function in tensorflow::ConditionalAccumulatorBase32 Status ConditionalAccumulatorBase::MatchesNodeDef(const NodeDef& node_def) { in MatchesNodeDef()42 Status ConditionalAccumulatorBase::SetGlobalStep(int64 new_global_step) { in SetGlobalStep()64 void ConditionalAccumulatorBase::TryTakeGrad(int num_required, in TryTakeGrad()110 void ConditionalAccumulatorBase::Cancel( in Cancel()138 bool ConditionalAccumulatorBase::TryAttemptLocked( in TryAttemptLocked()168 void ConditionalAccumulatorBase::FlushUnlocked() { in FlushUnlocked()190 bool ConditionalAccumulatorBase::TakeGradLockedHelper(OpKernelContext* ctx, in TakeGradLockedHelper()
35 return [this](ConditionalAccumulatorBase** ret) { in GetCreator()74 ConditionalAccumulatorBase* accumulator) override { in CheckSignature()100 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 {40 : ConditionalAccumulatorBase(dtype, shape, name, reduction_type) {} 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"
1162 class ConditionalAccumulatorBase(object): class1241 class ConditionalAccumulator(ConditionalAccumulatorBase):1330 class SparseConditionalAccumulator(ConditionalAccumulatorBase):