Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_while_op.cc175 std::vector<int> op_input_output_to_lowered_node_; member in tensorflow::__anon9eb550b10111::LowerWhileHelper
212 op_input_output_to_lowered_node_.resize(num_loop_inputs_, -1); in LowerWhileHelper()
233 op_input_output_to_lowered_node_[i] = counter++; in InitializeInputOutputToLoweredNodeMap()
304 NodeOut(merge_nodes_[op_input_output_to_lowered_node_[i]], 0)); in CreateCondFuncCallNode()
332 Node* merge_node = merge_nodes_[op_input_output_to_lowered_node_[i]]; in CreateSwitchNodes()
356 NodeOut(switch_nodes_[op_input_output_to_lowered_node_[i]], 1)); in CreateBodyFuncCallNode()
396 .Input(NodeOut(switch_nodes_[op_input_output_to_lowered_node_[i]], in CreateExitNodes()
398 .Device(switch_nodes_[op_input_output_to_lowered_node_[i]] in CreateExitNodes()
400 .AssignedDevice(switch_nodes_[op_input_output_to_lowered_node_[i]] in CreateExitNodes()
449 Node* merge_node = merge_nodes_[op_input_output_to_lowered_node_[i]]; in CreateNextIterationNodes()
[all …]