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

[INTEL_MKL] Changed 4 unit tests because of the node label changes recently introduced by graph rewrite changes

Browse Source
This commit is contained in:
Xiaoming (Jason) Cui 2019-06-28 18:21:34 -07:00
parent 813049eb44
commit 5481cbf062
4 changed files with 8 additions and 8 deletions

10
tensorflow/core/kernels/mkl_fused_ops_test.cc
View File

@ -57,7 +57,7 @@ class CommonTestUtilities : public OpsTestBase {
.Input(FakeInput(dtype)) // Input
.Input(FakeInput(DT_UINT8)) // Mkl second tensor
.Attr("T", dtype)
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());
AddInputFromArray<T>(tensor.shape(), tensor.flat<T>());
@ -232,7 +232,7 @@ class MklFusedConv2DOpTest : public OpsTestBase {
.Attr("strides", {1, stride, stride, 1})
.Attr("padding", "SAME")
.Attr("fused_ops", fused_ops)
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());
@ -432,7 +432,7 @@ class FusedPadConvOpTest : public OpsTestBase {
.Attr("data_format", data_format)
.Attr("T", dtype)
.Attr("strides", {1, stride, stride, 1})
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());
@ -526,7 +526,7 @@ class FilterCacheTest : public OpsTestBase {
.Attr("is_filter_const", is_filter_const)
.Attr("T", dtype)
.Attr("strides", {1, stride, stride, 1})
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());
@ -724,7 +724,7 @@ class MklPadWithFusedConv2DOpTest : public OpsTestBase {
.Attr("strides", {1, stride, stride, 1})
.Attr("padding", "VALID")
.Attr("fused_ops", fused_ops)
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());

2
tensorflow/core/kernels/mkl_quantized_conv_ops_perchannel_test.cc
View File

@ -53,7 +53,7 @@ class ConvMklToTF : public OpsTestBase {
.Input(FakeInput(dtype)) // Input
.Input(FakeInput(DT_UINT8)) // MKL second tensor
.Attr("T", dtype)
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());
AddInputFromArray<T>(first.shape(), first.flat<T>());

2
tensorflow/core/kernels/mkl_quantized_conv_ops_test.cc
View File

@ -55,7 +55,7 @@ class ConvMklToTF : public OpsTestBase {
.Input(FakeInput(dtype)) // Input
.Input(FakeInput(DT_UINT8)) // MKL metadata tensor
.Attr("T", dtype)
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());
AddInputFromArray<T>(input.shape(), input.flat<T>());

2
tensorflow/core/kernels/mkl_quantized_pooling_ops_test.cc
View File

@ -47,7 +47,7 @@ class ConvMklToTF : public OpsTestBase {
.Input(FakeInput(dtype)) // Input
.Input(FakeInput(DT_UINT8)) // Mkl second tensor
.Attr("T", dtype)
.Attr("_kernel", "MklOp")
.Attr("_kernel", "MklLayoutDependentOp")
.Finalize(node_def()));
TF_EXPECT_OK(InitOp());
AddInputFromArray<T>(first.shape(), first.flat<T>());