Searched refs:curr_node (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | control_flow.cc | 103 const Node* curr_node = ready.front(); in BuildControlFlowInfo() local
105 const ControlFlowInfo& curr_info = (*info)[curr_node->id()]; in BuildControlFlowInfo()
110 if (IsExit(curr_node)) { in BuildControlFlowInfo()
118 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo()
130 parent_nodes[out->id()] = curr_node; in BuildControlFlowInfo()
142 FormatNodeForError(*curr_node), " is in frame '", frame_name, in BuildControlFlowInfo()
163 FormatNodeForError(*curr_node), " is in frame '", frame_name, in BuildControlFlowInfo()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 227 NodeDef* curr_node = node_map.GetNode(curr); in IdentityNTerminalPorts() local
228 if (curr_node == nullptr || in IdentityNTerminalPorts()
229 visited.find(curr_node->name()) != visited.end()) { in IdentityNTerminalPorts()
236 if (IsIdentityN(*curr_node)) { in IdentityNTerminalPorts()
239 int pos = NodePositionIfSameNode(curr, curr_node->name()); in IdentityNTerminalPorts()
241 to_visit.push_back(curr_node->input(pos)); in IdentityNTerminalPorts()
243 for (const string& input : curr_node->input()) { in IdentityNTerminalPorts()
251 for (const string& input : curr_node->input()) { in IdentityNTerminalPorts()
254 visited.emplace(curr_node->name()); in IdentityNTerminalPorts()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | immutable_executor_state.cc | 305 Node* curr_node = ready.front(); in BuildControlFlowInfo() local
306 int curr_id = curr_node->id(); in BuildControlFlowInfo()
310 if (IsEnter(curr_node)) { in BuildControlFlowInfo()
313 GetNodeAttr(curr_node->attrs(), "frame_name", &frame_name)); in BuildControlFlowInfo()
314 parent = curr_node; in BuildControlFlowInfo()
315 } else if (IsExit(curr_node)) { in BuildControlFlowInfo()
325 for (const Edge* out_edge : curr_node->out_edges()) { in BuildControlFlowInfo()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 470 for (const auto* curr_node : fetch_fanin_nodes) { in Init() local
471 auto& curr_node_state = GetNodeStateOrCreateIt(curr_node); in Init()
472 const string curr_node_device = DeviceName(curr_node); in Init()
474 if (IsRecv(*curr_node)) { in Init()
475 const auto& attr = curr_node->attr(); in Init()
487 for (const string& input : curr_node->input()) { in Init()
511 input_node_state.outputs[input_node_port_num].push_back(curr_node); in Init()
523 input_node_state.outputs[0].push_back(curr_node); in Init()
528 CreateSendRecv(input_node, curr_node, input_node, input_node_name, in Init()
547 feed_nodes.find(curr_node->name()) != feed_nodes.end(); in Init()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 282 const Node* curr_node = nullptr; in GetNodeWithHighestPriority() local
286 curr_node = n; in GetNodeWithHighestPriority()
290 return curr_node; in GetNodeWithHighestPriority()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | deadness_analysis.cc | 1284 Node* curr_node = ready.front(); in GetFrameBasedTopologicalOrder() local
1287 VLOG(4) << "Visiting " << curr_node->name(); in GetFrameBasedTopologicalOrder()
1288 order->push_back(curr_node); in GetFrameBasedTopologicalOrder()
1290 for (const Edge* out_edge : curr_node->out_edges()) { in GetFrameBasedTopologicalOrder()
1293 if (IsNextIteration(curr_node) && IsMerge(out)) { in GetFrameBasedTopologicalOrder()
|