Searched refs:resource_retval_to_arg (Results 1 – 1 of 1) sorted by relevance
182 std::map<int, int>* resource_retval_to_arg) { in CalculateRetvalRearrange() argument204 resource_retval_to_arg->insert(std::make_pair(i, src_index)); in CalculateRetvalRearrange()230 const std::map<int, int>& resource_retval_to_arg) { in RearrangeOutputEdges() argument243 TF_RET_CHECK(resource_retval_to_arg.find(src_output) != in RearrangeOutputEdges()244 resource_retval_to_arg.end()); in RearrangeOutputEdges()248 n->input_edge(resource_retval_to_arg.at(src_output), &input_edge)); in RearrangeOutputEdges()431 std::map<int, int> resource_retval_to_arg, retval_index_mapping; in MaybeRewriteIfNode() local448 fbody->ret_nodes, &retval_index_mapping, &resource_retval_to_arg)); in MaybeRewriteIfNode()486 TF_RET_CHECK(resource_retval_to_arg.find(src_output) != in MaybeRewriteIfNode()487 resource_retval_to_arg.end()); in MaybeRewriteIfNode()[all …]