experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #30696 from Intel-tensorflow:bugfix/wenxizhu/TFDO-3199

Browse Source 
PiperOrigin-RevId: 258182674
This commit is contained in:
TensorFlower Gardener 2019-07-15 10:27:27 -07:00
commit 8df7ce02ba
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/graph/mkl_layout_pass_test.cc
View File

@ -2284,8 +2284,8 @@ TEST_F(MklLayoutPassTest, NodeRewrite_BiasAddGrad_Positive1) {
" attr { key: 'data_format' value { s: 'NCHW' } }"
" input: ['D'] }");
EXPECT_EQ(DoMklLayoutOptimizationPass(),
"A(Input);B(Input);C(MatMul);D(Zeta);E(BiasAddGrad)|"
"A->C;A->D:1;B->C:1;C->D;D->E");
"A(Input);B(Input);C(_MklMatMul);D(Zeta);E(BiasAddGrad)"
"|A->C;A->D:1;B->C:1;C->D;D->E");
}
// Check that we never rewrite BiasAddGrad.