Home
last modified time | relevance | path

Searched refs:while_ctx (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients.cc57 Status AddForwardLoopCounter(WhileContext* while_ctx, const Scope& scope, in AddForwardLoopCounter() argument
67 CondGraphBuilderFn cond_fn = [while_ctx](const Scope& scope, in AddForwardLoopCounter()
70 *output = ToOutput(while_ctx->cond_output()); in AddForwardLoopCounter()
75 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope, in AddForwardLoopCounter()
86 while_ctx->frame_name(), &outputs, in AddForwardLoopCounter()
96 Status AddBackPropLoopCounter(WhileContext* while_ctx, const Output& loop_count, in AddBackPropLoopCounter() argument
122 string frame_name = BackPropFrameName(while_ctx->frame_name()); in AddBackPropLoopCounter()
136 Status AddWhileGradientLoop(WhileContext* while_ctx, in AddWhileGradientLoop() argument
141 DCHECK_EQ(grad_inputs.size(), while_ctx->body_outputs().size()); in AddWhileGradientLoop()
142 DCHECK_EQ(while_ctx->body_inputs().size(), while_ctx->body_outputs().size()); in AddWhileGradientLoop()
[all …]
Dgradients.cc402 WhileContext* while_ctx = exit_node->while_ctx(); in ProcessWhileLoop() local
403 DCHECK(while_ctx != nullptr); in ProcessWhileLoop()
406 std::map<Node*, Output>& backprops = while_backprops_[while_ctx]; in ProcessWhileLoop()
413 if (backprops.size() < while_ctx->exit_nodes().size()) return Status::OK(); in ProcessWhileLoop()
418 scope_.NewSubScope(strings::StrCat(while_ctx->frame_name(), "_grad")); in ProcessWhileLoop()
420 for (Node* n : while_ctx->exit_nodes()) dy.push_back(backprops[n]); in ProcessWhileLoop()
422 TF_RETURN_IF_ERROR(AddWhileLoopGradient(while_ctx, while_scope, dy, &dx)); in ProcessWhileLoop()
426 DCHECK_EQ(dx.size(), while_ctx->enter_nodes().size()); in ProcessWhileLoop()
428 Node* enter_node = while_ctx->enter_nodes()[i]; in ProcessWhileLoop()
Dwhile_gradients.h34 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope,
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop_test.cc97 WhileContext* while_ctx; in TEST_F() local
102 ASSERT_TRUE(node->while_ctx() != nullptr) << i; in TEST_F()
104 while_ctx = node->while_ctx(); in TEST_F()
105 EXPECT_EQ(while_ctx->frame_name(), kFrameName); in TEST_F()
107 EXPECT_EQ(node->while_ctx(), while_ctx) << i; in TEST_F()
Dwhile_loop.cc234 WhileContext* while_ctx; in BuildWhileLoop() local
238 ToOutputTensors(body_outputs), &while_ctx)); in BuildWhileLoop()
243 (*outputs)[i].node()->set_while_ctx(while_ctx); in BuildWhileLoop()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h186 WhileContext* while_ctx() const { return while_ctx_; } in while_ctx() function
187 void set_while_ctx(WhileContext* while_ctx) { in set_while_ctx() argument
190 while_ctx_ = while_ctx; in set_while_ctx()