From 0d15ad80195ec5cd33f057068e34aa7e1dc2b783 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Thu, 5 Nov 2015 14:22:30 -0800 Subject: Updated the regressions tests that cover full reductions --- unsupported/test/cxx11_tensor_argmax.cpp | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'unsupported/test/cxx11_tensor_argmax.cpp') diff --git a/unsupported/test/cxx11_tensor_argmax.cpp b/unsupported/test/cxx11_tensor_argmax.cpp index 6eeecb717..482dfa7de 100644 --- a/unsupported/test/cxx11_tensor_argmax.cpp +++ b/unsupported/test/cxx11_tensor_argmax.cpp @@ -61,14 +61,14 @@ static void test_argmax_tuple_reducer() Tensor, 4, DataLayout> index_tuples(2,3,5,7); index_tuples = tensor.index_tuples(); - Tensor, 1, DataLayout> reduced(1); + Tensor, 0, DataLayout> reduced; DimensionList dims; reduced = index_tuples.reduce( dims, internal::ArgMaxTupleReducer>()); - Tensor maxi = tensor.maximum(); + Tensor maxi = tensor.maximum(); - VERIFY_IS_EQUAL(maxi(0), reduced(0).second); + VERIFY_IS_EQUAL(maxi(), reduced(0).second); array reduce_dims; for (int d = 0; d < 3; ++d) reduce_dims[d] = d; @@ -93,14 +93,14 @@ static void test_argmin_tuple_reducer() Tensor, 4, DataLayout> index_tuples(2,3,5,7); index_tuples = tensor.index_tuples(); - Tensor, 1, DataLayout> reduced(1); + Tensor, 0, DataLayout> reduced; DimensionList dims; reduced = index_tuples.reduce( dims, internal::ArgMinTupleReducer>()); - Tensor mini = tensor.minimum(); + Tensor mini = tensor.minimum(); - VERIFY_IS_EQUAL(mini(0), reduced(0).second); + VERIFY_IS_EQUAL(mini(), reduced(0).second); array reduce_dims; for (int d = 0; d < 3; ++d) reduce_dims[d] = d; @@ -123,7 +123,7 @@ static void test_simple_argmax() tensor = (tensor + tensor.constant(0.5)).log(); tensor(0,0,0,0) = 10.0; - Tensor tensor_argmax(1); + Tensor tensor_argmax; tensor_argmax = tensor.argmax(); @@ -144,7 +144,7 @@ static void test_simple_argmin() tensor = (tensor + tensor.constant(0.5)).log(); tensor(0,0,0,0) = -10.0; - Tensor tensor_argmin(1); + Tensor tensor_argmin; tensor_argmin = tensor.argmin(); -- cgit v1.2.3