This commit is contained in:
Yangqing Jia
2016-01-28 11:11:45 -08:00
parent c4e47630b1
commit 270c4e1ecd
4 changed files with 23 additions and 27 deletions

View File

@@ -10,13 +10,6 @@ typedef int TensorIndex;
#define BENCHMARK_RANGE(bench, lo, hi) \
BENCHMARK(bench)->Range(lo, hi)
template <typename... Args>
std::string StrCat(const Args... args) {
std::stringstream ss;
StrCatRecursive(ss, args...);
return ss.str();
}
using Eigen::Tensor;
using Eigen::TensorMap;
@@ -305,9 +298,9 @@ template <typename Device> class BenchmarkSuite {
}
size_t m_;
size_t k_;
size_t n_;
TensorIndex m_;
TensorIndex k_;
TensorIndex n_;
float* a_;
float* b_;
float* c_;