Searched refs:ConditionalAccumulatorBase (Results 1 – 13 of 13) sorted by relevance
71 ->template Delete<ConditionalAccumulatorBase>( in ~ConditionalAccumulatorBaseOp()83 typedef std::function<Status(ConditionalAccumulatorBase**)> Creator;106 ConditionalAccumulatorBase* accumulator; in SetAccumulatorHandle()109 ->template LookupOrCreate<ConditionalAccumulatorBase>( in SetAccumulatorHandle()135 ConditionalAccumulatorBase* accumulator; in Compute()143 ConditionalAccumulatorBase* accumulator) = 0;146 ConditionalAccumulatorBase* accumulator) = 0;149 ConditionalAccumulatorBase* accumulator) { in CheckSignature()169 ConditionalAccumulatorBase* accumulator) override { in Compute()198 ConditionalAccumulatorBase* accumulator; in ComputeAsync()[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()
34 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()39 ConditionalAccumulatorBase* accumulator) override { in Compute()70 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()95 ConditionalAccumulatorBase* accumulator) override { in CheckSignature()102 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()107 ConditionalAccumulatorBase* accumulator) override { in Compute()131 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()
35 return [this](ConditionalAccumulatorBase** ret) { in GetCreator()72 return [this](ConditionalAccumulatorBase** ret) { in GetCreator()93 TypeIndex::Make<ConditionalAccumulatorBase>())); in SetHandleToOutput()136 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()157 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()183 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()200 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()
35 return [this](ConditionalAccumulatorBase** ret) { in GetCreator()86 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()113 ConditionalAccumulatorBase* accumulator, in CheckSignature()124 ConditionalAccumulatorBase* accumulator) override { in GetExpectedInputs()
34 class TypedConditionalAccumulatorBase : public ConditionalAccumulatorBase {40 : ConditionalAccumulatorBase(dtype, shape, name, reduction_type) {} in TypedConditionalAccumulatorBase()
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"
1240 class ConditionalAccumulatorBase(object): class1320 class ConditionalAccumulator(ConditionalAccumulatorBase):1412 class SparseConditionalAccumulator(ConditionalAccumulatorBase):