Searched defs:cond_arg_nodes (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | functionalize_control_flow.cc | 639 StatusOr<Node*> ConvertToXlaIf(const CondArgNodes& cond_arg_nodes, 644 StatusOr<Node*> BuildAndAddXlaIfOp(const CondArgNodes& cond_arg_nodes, 650 Status ExtractBody(const CondArgNodes& cond_arg_nodes, 656 Status AddInputEdges(const CondArgNodes& cond_arg_nodes, Node* predicate, 1104 CondArgNodes cond_arg_nodes; local 1111 input_index[in] = cond_arg_nodes.size(); 1112 cond_arg_nodes.emplace_back(in); 1114 cond_arg_nodes.at(input_index.at(in)).switches.push_back(switch_node); 1124 ConvertToXlaIf(cond_arg_nodes, ps, merge_nodes)); 1132 for (auto& kv : cond_arg_nodes) { 1145 BuildAndAddXlaIfOp( const CondArgNodes& cond_arg_nodes, const SwitchCluster& switch_cluster, const std::vector<Node*>& merge_nodes) argument 1216 ExtractBody(const CondArgNodes& cond_arg_nodes, const std::vector<Node*>& switches, const std::vector<Node*>& merge_nodes, int input_edge, Graph* body) argument 1266 AddInputEdges(const CondArgNodes& cond_arg_nodes, Node* predicate, Node* if_node) argument 1315 ConvertToXlaIf( const CondArgNodes& cond_arg_nodes, const SwitchCluster& switch_cluster, const std::vector<Node*>& merge_nodes) argument [all...] |
Completed in 107 milliseconds