clang-format off
| 205 | |
| 206 | // clang-format off |
| 207 | TEST_F(MklLayoutPassTest, Basic) { |
| 208 | InitGraph( |
| 209 | "node { name: 'A' op: 'Input'}" |
| 210 | "node { name: 'B' op: 'Input'}" |
| 211 | "node { name: 'C' op: 'Zeta' attr { key: 'T' value { type: DT_FLOAT } }" |
| 212 | " input: ['A', 'B'] }" |
| 213 | "node { name: 'D' op: 'Zeta' attr { key: 'T' value { type: DT_FLOAT } }" |
| 214 | " input: ['A', 'B'] }"); |
| 215 | EXPECT_EQ(DoMklLayoutOptimizationPass(), |
| 216 | "A(Input);B(Input);C(Zeta);D(Zeta)|" |
| 217 | "A->C;A->D;B->C:1;B->D:1"); |
| 218 | } |
| 219 | |
| 220 | // Test set 1: Conv2D + AddBias |
| 221 |
nothing calls this directly
no test coverage detected