Searched refs:while_ctx (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/cc/framework/ |
D | while_gradients.cc | 57 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() 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() 160 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope, in AddWhileGradientLoop() [all …]
|
D | gradients.cc | 403 WhileContext* while_ctx = exit_node->while_ctx(); in ProcessWhileLoop() local 404 DCHECK(while_ctx != nullptr); in ProcessWhileLoop() 407 std::map<Node*, Output>& backprops = while_backprops_[while_ctx]; in ProcessWhileLoop() 414 if (backprops.size() < while_ctx->exit_nodes().size()) return Status::OK(); in ProcessWhileLoop() 419 scope_.NewSubScope(strings::StrCat(while_ctx->frame_name(), "_grad")); in ProcessWhileLoop() 421 for (Node* n : while_ctx->exit_nodes()) dy.push_back(backprops[n]); in ProcessWhileLoop() 423 TF_RETURN_IF_ERROR(AddWhileLoopGradient(while_ctx, while_scope, dy, &dx)); in ProcessWhileLoop() 427 DCHECK_EQ(dx.size(), while_ctx->enter_nodes().size()); in ProcessWhileLoop() 429 Node* enter_node = while_ctx->enter_nodes()[i]; in ProcessWhileLoop()
|
D | while_gradients.h | 34 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope,
|
/external/tensorflow/tensorflow/cc/ops/ |
D | while_loop_test.cc | 97 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()
|
D | while_loop.cc | 234 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/ |
D | graph.h | 234 WhileContext* while_ctx() const { return while_ctx_; } in while_ctx() function 235 void set_while_ctx(WhileContext* while_ctx) { in set_while_ctx() argument 238 while_ctx_ = while_ctx; in set_while_ctx()
|