Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dtransform_graph_function.cc51 std::unordered_map<string, string> node_name_to_control_ret; in TransformGraphFunction() local
59 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in TransformGraphFunction()
105 node_name_to_control_ret.emplace(control_ret, control_ret); in TransformGraphFunction()
109 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in TransformGraphFunction()
172 [&node_name_to_control_ret](const Node* n) -> absl::optional<string> { in TransformGraphFunction()
173 const auto it = node_name_to_control_ret.find(n->name()); in TransformGraphFunction()
174 if (it != node_name_to_control_ret.end()) in TransformGraphFunction()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime.cc869 std::unordered_map<string, string> node_name_to_control_ret; in InstantiateMultiDevice() local
882 node_name_to_control_ret.emplace(control_ret, control_ret); in InstantiateMultiDevice()
886 node_name_to_control_ret.emplace(control_ret.second, control_ret.first); in InstantiateMultiDevice()
1016 [&node_name_to_control_ret](const Node* n) -> absl::optional<string> { in InstantiateMultiDevice()
1017 const auto it = node_name_to_control_ret.find(n->name()); in InstantiateMultiDevice()
1018 return it != node_name_to_control_ret.end() in InstantiateMultiDevice()