diff options
author | 2018-03-08 07:31:01 +0000 | |
---|---|---|
committer | 2018-03-07 23:31:01 -0800 | |
commit | 9d867e0c34ea34ac74ebdab2cdcfc5b8c61fed25 (patch) | |
tree | eebc7f55ffc6e642cd4984ada9ed855da3ff3e8e | |
parent | d2d74f0d8256730955d3015861161d3b63eccb3a (diff) |
Add header and macros to allow these tests to be disabled in a manifest (#16977)
-rw-r--r-- | tensorflow/compiler/xla/tests/convolution_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/tests/convolution_test.cc b/tensorflow/compiler/xla/tests/convolution_test.cc index 99640f5bb5..72715398de 100644 --- a/tensorflow/compiler/xla/tests/convolution_test.cc +++ b/tensorflow/compiler/xla/tests/convolution_test.cc @@ -723,7 +723,7 @@ INSTANTIATE_TEST_CASE_P( ); #endif -TEST_F(ConvolutionTest, Convolve_bf16_1x1x1x2_1x1x1x2_Valid) { +XLA_TEST_F(ConvolutionTest, Convolve_bf16_1x1x1x2_1x1x1x2_Valid) { ComputationBuilder builder(client_, TestName()); Shape input_shape = ShapeUtil::MakeShape(BF16, {1, 1, 1, 2}); Shape filter_shape = ShapeUtil::MakeShape(BF16, {1, 1, 1, 2}); |