| 35 | PartialTensorShape Unknown() { return PartialTensorShape(); } |
| 36 | |
| 37 | OpDef MakeOpDef(int num_inputs, int num_outputs) { |
| 38 | OpRegistrationData op_reg_data; |
| 39 | OpDefBuilder b("dummy"); |
| 40 | for (int i = 0; i < num_inputs; ++i) { |
| 41 | b.Input(strings::StrCat("i", i, ": float")); |
| 42 | } |
| 43 | for (int i = 0; i < num_outputs; ++i) { |
| 44 | b.Output(strings::StrCat("o", i, ": float")); |
| 45 | } |
| 46 | CHECK(b.Attr("foo:string").Finalize(&op_reg_data).ok()); |
| 47 | return op_reg_data.op_def; |
| 48 | } |
| 49 | |
| 50 | } // namespace |
| 51 |