| 100 | } |
| 101 | |
| 102 | static OpDef MakeOpDef(int num_inputs, int num_outputs) { |
| 103 | OpRegistrationData op_reg_data; |
| 104 | OpDefBuilder b("dummy"); |
| 105 | for (int i = 0; i < num_inputs; ++i) { |
| 106 | b.Input(strings::StrCat("i", i, ": float")); |
| 107 | } |
| 108 | for (int i = 0; i < num_outputs; ++i) { |
| 109 | b.Output(strings::StrCat("o", i, ": float")); |
| 110 | } |
| 111 | CHECK(b.Attr("foo:string").Finalize(&op_reg_data).ok()); |
| 112 | return op_reg_data.op_def; |
| 113 | } |
| 114 | |
| 115 | TEST_F(ShapeInferenceTest, DimensionOrConstant) { |
| 116 | NodeDef def; |