Searched refs:AddForwardLoopCounter (Results 1 – 2 of 2) sorted by relevance
57 Status AddForwardLoopCounter(WhileContext* while_ctx, const Scope& scope, in AddForwardLoopCounter() function183 TF_RETURN_IF_ERROR(AddForwardLoopCounter( in AddWhileLoopGradient()
706 cnt, forward_index = forward_ctxt.AddForwardLoopCounter(outer_grad_state)2541 def AddForwardLoopCounter(self, outer_grad_state): member in WhileContext