| 85 | }; |
| 86 | |
| 87 | TEST_F(OpDefBuilderTest, Attr) { |
| 88 | ExpectSuccess(b().Attr("a:string"), "attr: { name: 'a' type: 'string' }"); |
| 89 | ExpectSuccess(b().Attr("A: int"), "attr: { name: 'A' type: 'int' }"); |
| 90 | ExpectSuccess(b().Attr("a1 :float"), "attr: { name: 'a1' type: 'float' }"); |
| 91 | ExpectSuccess(b().Attr("a_a : bool"), "attr: { name: 'a_a' type: 'bool' }"); |
| 92 | ExpectSuccess(b().Attr("aB : type"), "attr: { name: 'aB' type: 'type' }"); |
| 93 | ExpectSuccess(b().Attr("aB_3\t: shape"), |
| 94 | "attr: { name: 'aB_3' type: 'shape' }"); |
| 95 | ExpectSuccess(b().Attr("t: tensor"), "attr: { name: 't' type: 'tensor' }"); |
| 96 | ExpectSuccess(b().Attr("XYZ\t:\tlist(type)"), |
| 97 | "attr: { name: 'XYZ' type: 'list(type)' }"); |
| 98 | ExpectSuccess(b().Attr("f: func"), "attr { name: 'f' type: 'func'}"); |
| 99 | } |
| 100 | |
| 101 | TEST_F(OpDefBuilderTest, AttrFailure) { |
| 102 | ExpectFailure( |
nothing calls this directly
no test coverage detected