Skip to content

Commit bb30dfc

Browse files
ezhulenevtensorflower-gardener
authored andcommitted
Add benchmarks comparing Mkl vs Default Conv2D ops.
PiperOrigin-RevId: 213346439
1 parent 55581a5 commit bb30dfc

3 files changed

Lines changed: 430 additions & 0 deletions

File tree

tensorflow/core/kernels/BUILD

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ load(
3030
"//tensorflow:tensorflow.bzl",
3131
"if_android",
3232
"tf_cc_test",
33+
"tf_cc_test_mkl",
3334
"tf_cc_tests",
3435
"tf_cc_binary",
3536
"tf_copts",
@@ -6228,6 +6229,26 @@ tf_mkl_kernel_library(
62286229
] + mkl_deps(),
62296230
)
62306231

6232+
tf_cc_test_mkl(
6233+
name = "mkl_conv_ops_test",
6234+
size = "small",
6235+
srcs = ["mkl_conv_ops_test.cc"],
6236+
deps = [
6237+
":ops_testutil",
6238+
":ops_util",
6239+
"//tensorflow/cc:cc_ops",
6240+
"//tensorflow/core:core_cpu",
6241+
"//tensorflow/core:framework",
6242+
"//tensorflow/core:framework_internal",
6243+
"//tensorflow/core:lib",
6244+
"//tensorflow/core:protos_all_cc",
6245+
"//tensorflow/core:tensorflow",
6246+
"//tensorflow/core:test",
6247+
"//tensorflow/core:test_main",
6248+
"//tensorflow/core:testlib",
6249+
],
6250+
)
6251+
62316252
tf_mkl_kernel_library(
62326253
name = "mkl_tfconv_op",
62336254
prefix = "mkl_tfconv",

0 commit comments

Comments
 (0)