| 111 | REGISTER_OP("_Mkl_Input").Output("o: uint8").SetIsStateful(); |
| 112 | |
| 113 | TEST_F(MklToTfConversionPass, Basic) { |
| 114 | InitGraph( |
| 115 | "node { name: 'A' op: 'Float_Input'}" |
| 116 | "node { name: 'B' op: 'Float_Input'}" |
| 117 | "node { name: 'C' op: 'Mul' attr { key: 'T' value { type: DT_FLOAT } }" |
| 118 | " input: ['A', 'B'] }" |
| 119 | "node { name: 'D' op: 'Mul' attr { key: 'T' value { type: DT_FLOAT } }" |
| 120 | " input: ['A', 'B'] }"); |
| 121 | EXPECT_EQ(DoRunMklToTfConversionPass(), |
| 122 | "A(Float_Input);B(Float_Input);C(Mul);D(Mul)|" |
| 123 | "A->C;A->D;B->C:1;B->D:1"); |
| 124 | } |
| 125 | |
| 126 | // MklConv2D followed by Non-Mkl layer |
| 127 | // C=MklConv2D(A,M,B,N); E=Sub(C,D) (for interleaved ordering) |
nothing calls this directly
no test coverage detected