Skip to content

Commit c5238fb

Browse files
hawkinsptensorflower-gardener
authored andcommitted
[TF:XLA] Implement 3d pooling ops in the TF/XLA bridge.
Change: 149538112
1 parent d357303 commit c5238fb

File tree

5 files changed

+777
-144
lines changed

5 files changed

+777
-144
lines changed

tensorflow/compiler/tests/BUILD

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,21 @@ tf_xla_py_test(
195195
],
196196
)
197197

198+
tf_xla_py_test(
199+
name = "pooling_ops_3d_test",
200+
size = "medium",
201+
srcs = ["pooling_ops_3d_test.py"],
202+
shard_count = 10,
203+
deps = [
204+
":xla_test",
205+
"//tensorflow/python:array_ops",
206+
"//tensorflow/python:framework_for_generated_wrappers",
207+
"//tensorflow/python:nn_ops",
208+
"//tensorflow/python:nn_ops_gen",
209+
"//tensorflow/python:platform_test",
210+
],
211+
)
212+
198213
tf_xla_py_test(
199214
name = "random_ops_test",
200215
size = "small",

0 commit comments

Comments
 (0)