| 51 | public: |
| 52 | // To specify an output to be consumed by one of the Input() methods below. |
| 53 | struct NodeOut { |
| 54 | NodeOut(StringPiece n, int i, DataType dt); |
| 55 | NodeOut(); // uninitialized, call Reset() before use. |
| 56 | void Reset(StringPiece n, int i, DataType dt); |
| 57 | string node; |
| 58 | int index; |
| 59 | DataType data_type; |
| 60 | }; |
| 61 | |
| 62 | // Specify the name and the Op (either via an OpDef or the name of |
| 63 | // the Op plus a registry) for the NodeDef. Other fields are |