Searched refs:fanout_node (Results 1 – 2 of 2) sorted by relevance
480 const NodeDef& fanout_node = graph_view.graph()->node(fanout_idx); in GetStackPushNodesToConvert() local481 VLOG(1) << "Fanout " << fanout_idx << " : " << fanout_node.name(); in GetStackPushNodesToConvert()482 if (IsStackPushOp(fanout_node)) { in GetStackPushNodesToConvert()487 if (graph_view.HasNode(fanout_node.input(0))) { in GetStackPushNodesToConvert()488 const NodeDef* stack_node = graph_view.GetNode(fanout_node.input(0)); in GetStackPushNodesToConvert()501 } else if (IsStackOp(fanout_node) || IsStackCloseOp(fanout_node) || in GetStackPushNodesToConvert()502 op_types_to_traverse.find(fanout_node.op()) != in GetStackPushNodesToConvert()505 } else if (!IsStackPopOp(fanout_node) || in GetStackPushNodesToConvert()507 nodes_to_preserve.find(fanout_node.name()) != in GetStackPushNodesToConvert()
1301 for (const NodeDef* fanout_node : transitive_fanout) { in RelaxAllocatorConstraints() local1303 (IsSend(*fanout_node) || in RelaxAllocatorConstraints()1304 CrossesTaskOrCpuGpuBoundary(*fanout_node, assign_node))) { in RelaxAllocatorConstraints()1308 if (optimized_nodes.find(fanout_node) == optimized_nodes.end() && in RelaxAllocatorConstraints()1309 IsAssign(*fanout_node)) { in RelaxAllocatorConstraints()1310 assign_nodes_in_fanout.push_back(fanout_node); in RelaxAllocatorConstraints()