Searched refs:BenchmarkSuite (Results 1 – 6 of 6) sorted by relevance
| /external/eigen/bench/tensors/ |
| D | tensor_benchmarks_fp16_gpu.cu | 15 BenchmarkSuite<Eigen::GpuDevice, Eigen::half> suite(device, N); \
45 BenchmarkSuite<Eigen::GpuDevice, Eigen::half> suite(device, D1, D2, D3); \
64 BenchmarkSuite<Eigen::GpuDevice, Eigen::half> suite(device, N); \
|
| D | tensor_benchmarks_gpu.cu | 15 BenchmarkSuite<Eigen::GpuDevice, float> suite(device, N); \
45 BenchmarkSuite<Eigen::GpuDevice, float> suite(device, D1, D2, D3); \
64 BenchmarkSuite<Eigen::GpuDevice, float> suite(device, N); \
|
| D | tensor_benchmarks_cpu.cc | 16 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, N); \
88 BenchmarkSuite<Eigen::DefaultDevice, float> suite(device, D1, D2, D3); \
92 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, D1, D2, D3); \
141 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, N); \
|
| D | contraction_benchmarks_cpu.cc | 18 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, D1, D2, D3); \
|
| D | tensor_benchmarks_sycl.cc | 31 BenchmarkSuite<Eigen::SyclDevice, float> suite(device, N); \
|
| D | tensor_benchmarks.h | 18 template <typename Device, typename T> class BenchmarkSuite {
20 BenchmarkSuite(const Device& device, size_t m, size_t k, size_t n) in BenchmarkSuite() function
25 BenchmarkSuite(const Device& device, size_t m) in BenchmarkSuite() function
30 ~BenchmarkSuite() { in ~BenchmarkSuite()
|