Home
last modified time | relevance | path

Searched refs:AddWhileLoopGradient (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dwhile_gradients.h34 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope,
H A Dwhile_gradients.cc179 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope, in AddWhileLoopGradient() function
H A Dgradients.cc423 TF_RETURN_IF_ERROR(AddWhileLoopGradient(while_ctx, while_scope, dy, &dx)); in ProcessWhileLoop()