Searched defs:out_node (Results 1 – 6 of 6) sorted by relevance
82 for (const auto& out_node : out_nodes) { in ReplaceReduce() local94 for (const auto& out_node : out_nodes) { in ReplaceReduce() local150 for (const auto& out_node : out_nodes) { in ReplaceBroadcast() local191 for (const auto& out_node : out_nodes_map[send_dev]) { in ReplaceBroadcast() local234 for (const auto& out_node : out_nodes_map[recv_dev]) { in ReplaceBroadcast() local
220 Node** out_node) { in AddNode()237 Node** out_node) { in AddNode()257 Node** out_node) { in AddNode()276 Node** out_node) { in AddNode()
318 Node** out_node) override { in AddNode()
351 for (const auto out_node : loop_switch_node->out_nodes()) { in AddNoOpAfterLastIteration() local
1599 std::shared_ptr<Node>* out_node) { in AddNode()
3870 NodeDef* out_node = optimized_graph->add_node(); in AddQuantizedMatMulMinMaxOutConstNodes() local