| 147 | REGISTER_OP("TestMul").Input("a: float").Input("b: float").Output("o: float"); |
| 148 | |
| 149 | TEST_F(SubgraphTest, Targets1) { |
| 150 | ExpectOK( |
| 151 | "node { name: 'W1' op: 'TestParams' }" |
| 152 | "node { name: 'W2' op: 'TestParams' }" |
| 153 | "node { name: 'input' op: 'TestInput' }" |
| 154 | "node { name: 't1' op: 'TestMul' input: [ 'W1', 'input:1' ] }" |
| 155 | "node { name: 't2' op: 'TestMul' input: [ 'W2', 't1' ] }" |
| 156 | "node { name: 't3_a' op: 'TestRelu' input: 't2' }" |
| 157 | "node { name: 't3_b' op: 'TestRelu' input: 't2' }"); |
| 158 | EXPECT_EQ("OK", Subgraph("", "", "t1")); |
| 159 | ExpectNodes("W1,input,t1"); |
| 160 | } |
| 161 | |
| 162 | TEST_F(SubgraphTest, Targets2) { |
| 163 | ExpectOK( |
nothing calls this directly
no test coverage detected