diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
-rw-r--r-- | tensorflow/core/kernels/BUILD | 113 |
1 files changed, 72 insertions, 41 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 9f516efd71..9c47d520d9 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -27,6 +27,7 @@ load( "tf_copts", "tf_opts_nortti_if_android", "tf_kernel_library", + "tf_mkl_kernel_library", "cc_header_only_library", ) load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") @@ -2241,6 +2242,12 @@ tf_kernel_library( tf_kernel_library( name = "matmul_op", + srcs = [ + "matmul_op.cc", + ] + if_mkl([ + "mkl_matmul_op.cc", + ]), + hdrs = ["matmul_op.h"], defines = select({ ":xsmm": [ "TENSORFLOW_USE_LIBXSMM", @@ -2248,13 +2255,14 @@ tf_kernel_library( ], "//conditions:default": [], }), - prefix = "matmul_op", deps = MATH_DEPS + select({ ":xsmm": [ "@libxsmm_archive//:xsmm_avx", ], "//conditions:default": [], - }), + }) + if_mkl([ + "//third_party/mkl:intel_binary_blob", + ]), ) tf_kernel_library( @@ -2770,6 +2778,7 @@ tf_kernel_library( "cudnn_pooling_gpu.h", "fractional_pool_common.h", "maxpooling_op.h", + "pooling_ops_3d.h", "pooling_ops_common.h", ], gpu_srcs = [ @@ -2780,6 +2789,8 @@ tf_kernel_library( "maxpooling_op_gpu.h", "pooling_ops_common.h", "pooling_ops_common_gpu.h", + "pooling_ops_3d_gpu.h", + "pooling_ops_3d_gpu.cu.cc", ], deps = [ ":conv_2d", @@ -4468,49 +4479,69 @@ tf_cc_test( ], ) -if_mkl( - tf_kernel_library( - name = "mkl_matmul_op", - prefix = "mkl_matmul", - deps = [ - ":math", - "//third_party/mkl:intel_binary_blob", - ], - ), +tf_mkl_kernel_library( + name = "mkl_conv_op", + prefix = "mkl_conv", + deps = [ + ":bounds_check", + ":conv_ops", + ":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_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", - ], - ), +tf_mkl_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", + ], ) -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", - ], - ), +tf_mkl_kernel_library( + name = "mkl_pooling_ops", + srcs = [ + "mkl_avgpooling_op.cc", + "mkl_maxpooling_op.cc", + "mkl_pooling_ops_common.cc", + ], + hdrs = ["mkl_pooling_ops_common.h"], + deps = [ + ":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", + ], +) + +tf_mkl_kernel_library( + name = "mkl_relu_op", + prefix = "mkl_relu", + 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", + ], ) # ----------------------------------------------------------------------------- |