diff options
author | 2017-07-25 21:35:55 -0700 | |
---|---|---|
committer | 2017-07-25 21:40:14 -0700 | |
commit | dbd0f2e2d46044c90e27290f4aa35ebccba2e557 (patch) | |
tree | eb97b9f69dbbe4dd9bf22ae6eb4ca249f2c393ed /tensorflow/compiler/xla/BUILD | |
parent | 19290f0567366bcdd3eb7adfe0f830a053cc1314 (diff) |
[XLA] Use HloEvaluator for convolution in reference_util.
PiperOrigin-RevId: 163164566
Diffstat (limited to 'tensorflow/compiler/xla/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/BUILD | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/BUILD b/tensorflow/compiler/xla/BUILD index e0a03a78f1..ba90b13b38 100644 --- a/tensorflow/compiler/xla/BUILD +++ b/tensorflow/compiler/xla/BUILD @@ -563,6 +563,9 @@ cc_library( ":xla_data_proto", "//tensorflow/compiler/xla/client:computation_builder", "//tensorflow/compiler/xla/client:padding", + "//tensorflow/compiler/xla/service:hlo", + "//tensorflow/compiler/xla/service:hlo_evaluator", + "//tensorflow/compiler/xla/service:shape_inference", "//tensorflow/compiler/xla/service/cpu:runtime_single_threaded_matmul", "//tensorflow/core:lib", ], |