Searched refs:small_model_gemm_sizes (Results 1 – 1 of 1) sorted by relevance
308 const int small_model_gemm_sizes[] = { in benchmark_small_model() local311 assert(sizeof(small_model_gemm_sizes) % in benchmark_small_model()312 (3 * sizeof(small_model_gemm_sizes[0])) == in benchmark_small_model()315 sizeof(small_model_gemm_sizes) / (3 * sizeof(small_model_gemm_sizes[0])); in benchmark_small_model()319 small_model_gemms[i].rows = small_model_gemm_sizes[3 * i + 1]; in benchmark_small_model()320 small_model_gemms[i].depth = small_model_gemm_sizes[3 * i + 2]; in benchmark_small_model()321 small_model_gemms[i].cols = small_model_gemm_sizes[3 * i + 0]; in benchmark_small_model()