Searched refs:enter_nodes (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/graph/ |
D | while_context.cc | 21 std::vector<Node*> enter_nodes, in WhileContext() argument 27 enter_nodes_(std::move(enter_nodes)), in WhileContext()
|
D | while_context.h | 37 WhileContext(StringPiece frame_name, std::vector<Node*> enter_nodes, 43 const std::vector<Node*>& enter_nodes() const { return enter_nodes_; } in enter_nodes() function
|
D | graph.cc | 850 std::vector<Node*> enter_nodes, in AddWhileContext() argument 858 WhileContext(frame_name, std::move(enter_nodes), std::move(exit_nodes), in AddWhileContext()
|
D | graph.h | 695 Status AddWhileContext(StringPiece frame_name, std::vector<Node*> enter_nodes,
|
/external/tensorflow/tensorflow/cc/framework/ |
D | gradients.cc | 427 DCHECK_EQ(dx.size(), while_ctx->enter_nodes().size()); in ProcessWhileLoop() 429 Node* enter_node = while_ctx->enter_nodes()[i]; in ProcessWhileLoop()
|