Searched defs:cond_arg_nodes (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_control_flow.cc639 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