Searched defs:out_node (Results 1 – 10 of 10) 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()
388 for (const auto out_node : switch_node->out_nodes()) { in AddNoOpAfterLastIteration() local
319 Node** out_node) override { in AddNode()
416 auto out_node = out->node(); in replaceListOutputWithTuple() local
700 auto out_node = in clamp_node_creator() local
722 def make_kernel_render(out_node): argument
1192 Node* out_node = out_edge->dst(); in SelectGraphToExecute() local
2130 std::shared_ptr<Node>* out_node) { in AddNode()
3929 NodeDef* out_node = optimized_graph->add_node(); in AddQuantizedMatMulMinMaxOutConstNodes() local