| 314 | } |
| 315 | |
| 316 | int NumControlOutputs(const NodeDef& node, const NodeMap& node_map) { |
| 317 | int num_outputs = 0; |
| 318 | for (const NodeDef* output : node_map.GetOutputs(node.name())) { |
| 319 | for (const string& node_as_input : output->input()) { |
| 320 | if (!IsControlInput(node_as_input)) continue; |
| 321 | |
| 322 | TensorId tensor = ParseTensorName(node_as_input); |
| 323 | if (tensor.node() == node.name()) { |
| 324 | ++num_outputs; |
| 325 | } |
| 326 | } |
| 327 | } |
| 328 | return num_outputs; |
| 329 | } |
| 330 | |
| 331 | int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map) { |
| 332 | int num_outputs = 0; |
no test coverage detected