diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
| -rw-r--r-- | tensorflow/core/kernels/BUILD | 63 |
1 files changed, 60 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 9740f96a6d..3b79d4c3db 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -688,8 +688,15 @@ tf_kernel_library( tf_kernel_library( name = "transpose_op", - prefix = "transpose_op", - deps = ARRAY_DEPS, + srcs = [ + "transpose_op.cc", + ] + if_mkl([ + "mkl_transpose_op.cc", + ]), + hdrs = ["transpose_op.h"], + deps = ARRAY_DEPS + if_mkl([ + "//third_party/mkl:intel_binary_blob", + ]), ) tf_kernel_library( @@ -1735,6 +1742,22 @@ tf_cuda_cc_test( ], ) +tf_cuda_cc_test( + name = "resize_benchmark_test", + srcs = ["resize_op_benchmark_test.cc"], + deps = [ + ":image", + ":ops_testutil", + ":ops_util", + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core:testlib", + ], +) + cc_library( name = "io", deps = [ @@ -4376,7 +4399,7 @@ tf_cc_test( if_mkl( tf_kernel_library( - name = "mkl_ops", + name = "mkl_matmul_op", prefix = "mkl_matmul", deps = [ ":math", @@ -4385,6 +4408,40 @@ if_mkl( ), ) +if_mkl( + tf_kernel_library( + name = "mkl_conv_op", + prefix = "mkl_conv", + deps = [ + ":bounds_check", + ":ops_util", + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", + "//tensorflow/core:nn_ops_op_lib", + "//third_party/mkl:intel_binary_blob", + ], + ), +) + +if_mkl( + tf_kernel_library( + name = "mkl_tfconv_op", + prefix = "mkl_tfconv", + deps = [ + ":bounds_check", + ":ops_util", + "//tensorflow/core:core_cpu", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", + "//tensorflow/core:nn_ops_op_lib", + "//third_party/mkl:intel_binary_blob", + ], + ), +) + # ----------------------------------------------------------------------------- # Google-internal targets. These must be at the end for syncrepo. |