| 349 | } |
| 350 | |
| 351 | int NumNonControlDataOutputs(const NodeDef& node, const NodeMap& node_map) { |
| 352 | int num_data_outputs = 0; |
| 353 | for (const NodeDef* output : node_map.GetOutputs(node.name())) { |
| 354 | if (IsShapeConsumer(*output)) continue; |
| 355 | |
| 356 | for (int i = 0; i < output->input_size(); ++i) { |
| 357 | const string& input = output->input(i); |
| 358 | if (!IsControlInput(input) && NodeName(input) == node.name()) { |
| 359 | ++num_data_outputs; |
| 360 | break; |
| 361 | } |
| 362 | } |
| 363 | } |
| 364 | return num_data_outputs; |
| 365 | } |
| 366 | |
| 367 | // Returns the data type in attribute `attr_name` of `node`. If that attribute |
| 368 | // doesn't exist, returns DT_INVALID. |