From d26ba01420ed9e4e1a55f22ca0e3034fdd3d3671 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 9 Dec 2019 15:09:57 -0800 Subject: [PATCH] [XLA] Fix reduce test to actually use the specified input PiperOrigin-RevId: 284644242 Change-Id: I84f2181660aeb80e75e42407aa49a4391a535433 --- tensorflow/compiler/xla/tests/reduce_hlo_test.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tensorflow/compiler/xla/tests/reduce_hlo_test.cc b/tensorflow/compiler/xla/tests/reduce_hlo_test.cc index d0d6a91e84b..2dc2fce10f9 100644 --- a/tensorflow/compiler/xla/tests/reduce_hlo_test.cc +++ b/tensorflow/compiler/xla/tests/reduce_hlo_test.cc @@ -106,7 +106,10 @@ XLA_TEST_P(ReduceWithLayoutTest, Reduce) { {-0.241772294, -0.245131493, -0.160247207}, {-0.179881215, -0.23383224, -0.121976733}}}}); - EXPECT_TRUE(RunAndCompareNoHloPasses(std::move(module), ErrorSpec(1e-5))); + auto reduce_input_relaid = + reduce_input.Relayout(reduce_input_shape->layout()); + EXPECT_TRUE(RunAndCompareNoHloPasses( + std::move(module), {&reduce_input_relaid}, ErrorSpec(1e-5))); } INSTANTIATE_TEST_CASE_P(ReduceWithLayoutTest_Instantiation,