Home
last modified time | relevance | path

Searched refs:merge_to_replacement (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc311 std::unordered_map<Node*, OutputTensor>* merge_to_replacement);
328 const std::unordered_map<Node*, OutputTensor>& merge_to_replacement);
334 std::unordered_map<Node*, OutputTensor>* merge_to_replacement);
847 const std::unordered_map<Node*, OutputTensor>& merge_to_replacement) { in AddInputEdges() argument
858 auto iter = merge_to_replacement.find(predicate_.node); in AddInputEdges()
859 if (iter == merge_to_replacement.end()) { in AddInputEdges()
884 std::unordered_map<Node*, OutputTensor>* merge_to_replacement) { in AddOutputEdges() argument
910 (*merge_to_replacement)[node] = OutputTensor{if_node_, i}; in AddOutputEdges()
923 std::unordered_map<Node*, OutputTensor>* merge_to_replacement) { in BuildAndReplace() argument
942 TF_RETURN_IF_ERROR(AddInputEdges(graph, *merge_to_replacement)); in BuildAndReplace()
[all …]