devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improved the performance of tensor padding

Browse Source
This commit is contained in:
Benoit Steiner
2016-05-25 11:43:08 -07:00
parent 58026905ae
commit 034aa3b2c0
2 changed files with 53 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
bench/tensors/tensor_benchmarks.h
View File

@@ -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) {