Searched refs:BM_FuncWithInputDimsGPU (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/bench/tensors/ |
| D | tensor_benchmarks_fp16_gpu.cu | 40 #define BM_FuncWithInputDimsGPU(FUNC, D1, D2, D3) \ macro
52 BM_FuncWithInputDimsGPU(contraction, N, N, N);
53 BM_FuncWithInputDimsGPU(contraction, 64, N, N);
54 BM_FuncWithInputDimsGPU(contraction, N, 64, N);
55 BM_FuncWithInputDimsGPU(contraction, N, N, 64);
|
| D | tensor_benchmarks_gpu.cu | 40 #define BM_FuncWithInputDimsGPU(FUNC, D1, D2, D3) \ macro
52 BM_FuncWithInputDimsGPU(contraction, N, N, N);
53 BM_FuncWithInputDimsGPU(contraction, 64, N, N);
54 BM_FuncWithInputDimsGPU(contraction, N, 64, N);
55 BM_FuncWithInputDimsGPU(contraction, N, N, 64);
|