From 0698453f709988f2c6fdd9c4bd0a3d7f0d771eab Mon Sep 17 00:00:00 2001 From: Hu Yuan Date: Tue, 8 Nov 2022 17:36:32 +0800 Subject: [PATCH] Fix the Mkl fused matmul node builder bug in unit test --- tensorflow/core/kernels/mkl_matmul_op_test.cc | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tensorflow/core/kernels/mkl_matmul_op_test.cc b/tensorflow/core/kernels/mkl_matmul_op_test.cc index 5c90159c3ee..70d38bd495d 100644 --- a/tensorflow/core/kernels/mkl_matmul_op_test.cc +++ b/tensorflow/core/kernels/mkl_matmul_op_test.cc @@ -254,10 +254,7 @@ static Graph* FusedMatMul(const string& kind, int m, int k, int n, .Attr("transpose_a", transpose_a) .Attr("transpose_b", transpose_b); - isDefault ? nodeBuilder : nodeBuilder.Attr("_kernel", "MklLayoutDependentOp") - .Input(not_mkl_shape) - .Input(not_mkl_shape) - .Input(args_not_mkl); + isDefault ? nodeBuilder : nodeBuilder.Attr("_kernel", "MklNameChangeOp"); TF_CHECK_OK(nodeBuilder.Finalize(g, nullptr));