aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/BUILD
diff options
context:
space:
mode:
authorGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-23 10:11:22 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-23 10:11:22 -0700
commitbb5fbbcb663c795dd7fc16e43a0eaaae53231fd9 (patch)
tree702f13489b23170ccf1149c01da4bdf501f5cf5a /tensorflow/core/kernels/BUILD
parent7d25d2d6c5db2269b6dba4cade6edaf7e8ddf6ba (diff)
parent33f57bd1311df97a25cd70784dfaafc8e44d07c4 (diff)
Merge pull request #21715 from hsgkim:volume_patches
PiperOrigin-RevId: 214177065
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
-rw-r--r--tensorflow/core/kernels/BUILD14
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
index 08245e6ea0..ab69925d04 100644
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@@ -217,6 +217,19 @@ tf_kernel_library(
],
)
+tf_kernel_library(
+ name = "extract_volume_patches_op",
+ prefix = "extract_volume_patches_op",
+ deps = [
+ ":bounds_check",
+ ":eigen_helpers",
+ ":ops_util",
+ "//tensorflow/core:framework",
+ "//tensorflow/core:lib",
+ "//third_party/eigen3",
+ ],
+)
+
cc_library(
name = "conv_3d",
hdrs = ["conv_3d.h"],
@@ -622,6 +635,7 @@ cc_library(
":diag_op",
":edit_distance_op",
":extract_image_patches_op",
+ ":extract_volume_patches_op",
":gather_nd_op",
":gather_op",
":guarantee_const_op",