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()
|