diff options
Diffstat (limited to 'tensorflow/core')
-rw-r--r-- | tensorflow/core/framework/common_shape_fns.cc | 2 | ||||
-rw-r--r-- | tensorflow/core/graph/mkl_layout_pass_test.cc | 6 |
2 files changed, 5 insertions, 3 deletions
diff --git a/tensorflow/core/framework/common_shape_fns.cc b/tensorflow/core/framework/common_shape_fns.cc index 5f34a75121..50403b4004 100644 --- a/tensorflow/core/framework/common_shape_fns.cc +++ b/tensorflow/core/framework/common_shape_fns.cc @@ -1335,7 +1335,7 @@ Status SliceHelper(InferenceContext* c, ShapeHandle begin_value, return Status::OK(); } -} // namespace +} // namespace Status SliceShape(InferenceContext* c) { ShapeHandle input = c->input(0); diff --git a/tensorflow/core/graph/mkl_layout_pass_test.cc b/tensorflow/core/graph/mkl_layout_pass_test.cc index cccef5a03a..7f96a18023 100644 --- a/tensorflow/core/graph/mkl_layout_pass_test.cc +++ b/tensorflow/core/graph/mkl_layout_pass_test.cc @@ -3523,8 +3523,10 @@ TEST_F(MklLayoutPassTest, NodeRewrite_Slice_DeviceTest) { " input: ['A', 'D'] }"); EXPECT_EQ(DoMklLayoutOptimizationPass(), "A(Input);B(Int32Input);C(Int32Input);" - "D(_MklSlice);DMT/_0(Const);DMT/_1(Const);DMT/_2(Const);E(Zeta)|A->D;A->E;" - "A:control->DMT/_0:control;A:control->DMT/_1:control;A:control->DMT/_2:control;" + "D(_MklSlice);DMT/_0(Const);DMT/_1(Const);DMT/" + "_2(Const);E(Zeta)|A->D;A->E;" + "A:control->DMT/_0:control;A:control->DMT/" + "_1:control;A:control->DMT/_2:control;" "B->D:1;C->D:2;D->E:1;DMT/_0->D:3;DMT/_1->D:4;DMT/_2->D:5"); } |