Searched refs:merge_to_replacement (Results 1 – 1 of 1) sorted by relevance
301 std::unordered_map<Node*, OutputTensor>* merge_to_replacement);318 const std::unordered_map<Node*, OutputTensor>& merge_to_replacement);324 std::unordered_map<Node*, OutputTensor>* merge_to_replacement);806 const std::unordered_map<Node*, OutputTensor>& merge_to_replacement) { in AddInputEdges() argument817 auto iter = merge_to_replacement.find(predicate_.node); in AddInputEdges()818 if (iter == merge_to_replacement.end()) { in AddInputEdges()843 std::unordered_map<Node*, OutputTensor>* merge_to_replacement) { in AddOutputEdges() argument869 (*merge_to_replacement)[node] = OutputTensor{if_node_, i}; in AddOutputEdges()882 std::unordered_map<Node*, OutputTensor>* merge_to_replacement) { in BuildAndReplace() argument901 TF_RETURN_IF_ERROR(AddInputEdges(graph, *merge_to_replacement)); in BuildAndReplace()[all …]