| /external/tensorflow/tensorflow/cc/framework/ |
| while_gradients.h | 28 // `while_ctx`. `grad_inputs` are the partial derivatives w.r.t. the loop
34 Status AddWhileLoopGradient(WhileContext* while_ctx, const Scope& scope,
|
| while_gradients.cc | 55 // while loop associated with `while_ctx`. The returned output yields the
57 Status AddForwardLoopCounter(WhileContext* while_ctx, const Scope& scope,
67 CondGraphBuilderFn cond_fn = [while_ctx](const Scope& scope,
70 *output = ToOutput(while_ctx->cond_output());
75 BodyGraphBuilderFn body_fn = [while_ctx](const Scope& scope,
86 while_ctx->frame_name(), &outputs,
95 // `while_ctx`.
96 Status AddBackPropLoopCounter(WhileContext* while_ctx, const Output& loop_count,
122 string frame_name = BackPropFrameName(while_ctx->frame_name());
131 // associated with `while_ctx`. `grad_inputs` are the partial derivative
[all...] |
| gradients.cc | 402 WhileContext* while_ctx = exit_node->while_ctx(); local
403 DCHECK(while_ctx != nullptr);
406 std::map<Node*, Output>& backprops = while_backprops_[while_ctx];
413 if (backprops.size() < while_ctx->exit_nodes().size()) return Status::OK();
418 scope_.NewSubScope(strings::StrCat(while_ctx->frame_name(), "_grad"));
420 for (Node* n : while_ctx->exit_nodes()) dy.push_back(backprops[n]);
422 TF_RETURN_IF_ERROR(AddWhileLoopGradient(while_ctx, while_scope, dy, &dx));
426 DCHECK_EQ(dx.size(), while_ctx->enter_nodes().size());
428 Node* enter_node = while_ctx->enter_nodes()[i]
[all...] |
| /external/tensorflow/tensorflow/cc/ops/ |
| while_loop_test.cc | 97 WhileContext* while_ctx; local
102 ASSERT_TRUE(node->while_ctx() != nullptr) << i;
104 while_ctx = node->while_ctx();
105 EXPECT_EQ(while_ctx->frame_name(), kFrameName);
107 EXPECT_EQ(node->while_ctx(), while_ctx) << i;
|
| while_loop.cc | 234 WhileContext* while_ctx; local
238 ToOutputTensors(body_outputs), &while_ctx));
240 // Set while_ctx for all exit nodes. We currently don't require knowing the
241 // while_ctx for any other nodes.
243 (*outputs)[i].node()->set_while_ctx(while_ctx);
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph.h | 186 WhileContext* while_ctx() const { return while_ctx_; } function in class:tensorflow::Node
187 void set_while_ctx(WhileContext* while_ctx) {
190 while_ctx_ = while_ctx;
655 // AddWhileContext() or Node::while_ctx(), but this manages the lifetime.
|
