Searched refs:fanout_node (Results 1 – 3 of 3) sorted by relevance
57 const NodeDef* fanout_node = &graph->node(fanout_node_index); in InferFromGraphViewT() local58 if (!node_to_frames_.contains(fanout_node)) { in InferFromGraphViewT()67 if (IsEnter(*fanout_node)) { in InferFromGraphViewT()69 AttrSlice(*fanout_node).Find("frame_name"); in InferFromGraphViewT()74 SummarizeNodeDef(*fanout_node)); in InferFromGraphViewT()91 node_to_frames_[fanout_node] = std::move(frame_ids); in InferFromGraphViewT()98 std::vector<int> frame_ids_fanout = node_to_frames_[fanout_node]; in InferFromGraphViewT()101 if (IsEnter(*fanout_node)) { in InferFromGraphViewT()111 " does not match frame ids for it's fanout ", fanout_node->name()); in InferFromGraphViewT()
479 const NodeDef& fanout_node = graph_view.graph()->node(fanout_idx); in GetStackPushNodesToConvert() local480 VLOG(1) << "Fanout " << fanout_idx << " : " << fanout_node.name(); in GetStackPushNodesToConvert()481 if (IsStackPushOp(fanout_node)) { in GetStackPushNodesToConvert()486 if (graph_view.HasNode(fanout_node.input(0))) { in GetStackPushNodesToConvert()487 const NodeDef* stack_node = graph_view.GetNode(fanout_node.input(0)); in GetStackPushNodesToConvert()500 } else if (IsStackOp(fanout_node) || IsStackCloseOp(fanout_node) || in GetStackPushNodesToConvert()501 op_types_to_traverse.find(fanout_node.op()) != in GetStackPushNodesToConvert()504 } else if (!IsStackPopOp(fanout_node) || in GetStackPushNodesToConvert()506 nodes_to_preserve.find(fanout_node.name()) != in GetStackPushNodesToConvert()
1347 for (const NodeDef* fanout_node : transitive_fanout) { in FindAssignNodesToRelax() local1349 (IsSend(*fanout_node) || in FindAssignNodesToRelax()1350 CrossesTaskOrCpuGpuBoundary(*fanout_node, assign_node))) { in FindAssignNodesToRelax()1354 if (optimized_nodes.find(fanout_node) == optimized_nodes.end() && in FindAssignNodesToRelax()1355 IsAssign(*fanout_node)) { in FindAssignNodesToRelax()1356 assign_nodes_in_fanout.push_back(fanout_node); in FindAssignNodesToRelax()