Searched refs:DIM1 (Results 1 – 3 of 3) sorted by relevance
| /external/eigen/bench/tensors/ |
| D | tensor_benchmarks_fp16_gpu.cu | 59 #define BM_FuncWithKernelDimsGPU(FUNC, DIM1, DIM2) \ argument
60 static void BM_##FUNC##_##DIM1##x##DIM2(int iters, int N) { \
66 suite.FUNC(iters, DIM1, DIM2); \
68 BENCHMARK_RANGE(BM_##FUNC##_##DIM1##x##DIM2, 128, 5000);
|
| D | tensor_benchmarks_gpu.cu | 59 #define BM_FuncWithKernelDimsGPU(FUNC, DIM1, DIM2) \ argument
60 static void BM_##FUNC##_##DIM1##x##DIM2(int iters, int N) { \
66 suite.FUNC(iters, DIM1, DIM2); \
68 BENCHMARK_RANGE(BM_##FUNC##_##DIM1##x##DIM2, 128, 5000);
|
| D | tensor_benchmarks_cpu.cc | 137 #define BM_FuncWithKernelDimsCPU(FUNC, DIM1, DIM2, THREADS) \ argument
138 static void BM_##FUNC##_##DIM1##x##DIM2##_##THREADS##T(int iters, int N) { \
142 suite.FUNC(iters, DIM1, DIM2); \
144 BENCHMARK_RANGE(BM_##FUNC##_##DIM1##x##DIM2##_##THREADS##T, 128, 5000);
|