| 138 | |
| 139 | protected: |
| 140 | NodeDef CreateNodeDef(const string& op_type, const DataTypeVector& inputs, |
| 141 | const string& device = "") { |
| 142 | NodeDefBuilder builder(op_type + "-op", op_type); |
| 143 | for (DataType dt : inputs) { |
| 144 | builder.Input(FakeInput(dt)); |
| 145 | } |
| 146 | builder.Device(device); |
| 147 | NodeDef node_def; |
| 148 | TF_CHECK_OK(builder.Finalize(&node_def)); |
| 149 | return node_def; |
| 150 | } |
| 151 | |
| 152 | void ExpectEqual(const string& what, const DataTypeVector& expected, |
| 153 | const DataTypeVector& observed) { |