diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-05-25 11:43:08 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-05-25 11:43:08 -0700 |
commit | 034aa3b2c0fa55cdf7d4bf052a067989218565c0 (patch) | |
tree | ab2a6f1ae3f8c4c18934cf2fd22b7c853665c1bd /bench/tensors/tensor_benchmarks.h | |
parent | 58026905ae4a608abac33f59a782beae590a8371 (diff) |
Improved the performance of tensor padding
Diffstat (limited to 'bench/tensors/tensor_benchmarks.h')
-rw-r--r-- | bench/tensors/tensor_benchmarks.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/bench/tensors/tensor_benchmarks.h b/bench/tensors/tensor_benchmarks.h index e0631b401..c2fb3dede 100644 --- a/bench/tensors/tensor_benchmarks.h +++ b/bench/tensors/tensor_benchmarks.h @@ -178,9 +178,14 @@ template <typename Device, typename T> class BenchmarkSuite { size_b[1] = m_; TensorMap<Tensor<T, 2>, Eigen::Aligned> B(b_, size_b); +#if defined(EIGEN_HAS_INDEX_LIST) + Eigen::IndexPairList<Eigen::type2indexpair<0, 0>, + Eigen::type2indexpair<2, 1> > paddings; +#else Eigen::array<Eigen::IndexPair<TensorIndex>, 2> paddings; paddings[0] = Eigen::IndexPair<TensorIndex>(0, 0); paddings[1] = Eigen::IndexPair<TensorIndex>(2, 1); +#endif StartBenchmarkTiming(); for (int iter = 0; iter < num_iters; ++iter) { |