| 193 | } |
| 194 | |
| 195 | void NodeDefBuilder::VerifyInputRef(const OpDef::ArgDef* input_arg, |
| 196 | DataType dt) { |
| 197 | if (input_arg->is_ref() && !IsRefType(dt)) { |
| 198 | errors_.push_back(strings::StrCat("Input '", input_arg->name(), "' passed ", |
| 199 | DataTypeString(dt), |
| 200 | " expected ref type")); |
| 201 | } |
| 202 | } |
| 203 | |
| 204 | NodeDefBuilder& NodeDefBuilder::ControlInput(StringPiece src_node) { |
| 205 | control_inputs_.emplace_back(src_node); |