diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-09-18 09:42:08 -0700 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2015-09-18 09:42:08 -0700 |
| commit | 13aee4463e2a9bcd13621453a5ca4c9922ea0dda (patch) | |
| tree | d6030e355344d9286b938f6c1f5de4b6b26154b1 /unsupported/test/cxx11_tensor_argmax.cpp | |
| parent | 58a6453d484602ace9c49302fb7370a270f4a051 (diff) | |
Cleaned up a test
Diffstat (limited to 'unsupported/test/cxx11_tensor_argmax.cpp')
| -rw-r--r-- | unsupported/test/cxx11_tensor_argmax.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/unsupported/test/cxx11_tensor_argmax.cpp b/unsupported/test/cxx11_tensor_argmax.cpp index 50904d5bc..6eeecb717 100644 --- a/unsupported/test/cxx11_tensor_argmax.cpp +++ b/unsupported/test/cxx11_tensor_argmax.cpp @@ -184,8 +184,8 @@ static void test_argmax_dim() tensor_argmax = tensor.argmax(dim); - VERIFY_IS_EQUAL(tensor_argmax.dimensions().TotalSize(), - size_t(2*3*5*7 / tensor.dimension(dim))); + VERIFY_IS_EQUAL(tensor_argmax.size(), + ptrdiff_t(2*3*5*7 / tensor.dimension(dim))); for (ptrdiff_t n = 0; n < tensor_argmax.size(); ++n) { // Expect max to be in the first index of the reduced dimension VERIFY_IS_EQUAL(tensor_argmax.data()[n], 0); @@ -208,7 +208,7 @@ static void test_argmax_dim() VERIFY_IS_EQUAL(tensor_argmax.size(), ptrdiff_t(2*3*5*7 / tensor.dimension(dim))); - for (ptrdiff_t n = 0; n < tensor_argmax.dimensions().TotalSize(); ++n) { + for (ptrdiff_t n = 0; n < tensor_argmax.size(); ++n) { // Expect max to be in the last index of the reduced dimension VERIFY_IS_EQUAL(tensor_argmax.data()[n], tensor.dimension(dim) - 1); } @@ -242,8 +242,8 @@ static void test_argmin_dim() tensor_argmin = tensor.argmin(dim); - VERIFY_IS_EQUAL(tensor_argmin.dimensions().TotalSize(), - size_t(2*3*5*7 / tensor.dimension(dim))); + VERIFY_IS_EQUAL(tensor_argmin.size(), + ptrdiff_t(2*3*5*7 / tensor.dimension(dim))); for (ptrdiff_t n = 0; n < tensor_argmin.size(); ++n) { // Expect min to be in the first index of the reduced dimension VERIFY_IS_EQUAL(tensor_argmin.data()[n], 0); @@ -266,7 +266,7 @@ static void test_argmin_dim() VERIFY_IS_EQUAL(tensor_argmin.size(), ptrdiff_t(2*3*5*7 / tensor.dimension(dim))); - for (ptrdiff_t n = 0; n < tensor_argmin.dimensions().TotalSize(); ++n) { + for (ptrdiff_t n = 0; n < tensor_argmin.size(); ++n) { // Expect min to be in the last index of the reduced dimension VERIFY_IS_EQUAL(tensor_argmin.data()[n], tensor.dimension(dim) - 1); } |