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