Searched refs:small_model_gemm_sizes (Results 1 – 1 of 1) sorted by relevance
316 const int small_model_gemm_sizes[] = { in benchmark_small_model() local319 assert(sizeof(small_model_gemm_sizes) % in benchmark_small_model()320 (3 * sizeof(small_model_gemm_sizes[0])) == in benchmark_small_model()323 sizeof(small_model_gemm_sizes) / (3 * sizeof(small_model_gemm_sizes[0])); in benchmark_small_model()327 small_model_gemms[i].rows = small_model_gemm_sizes[3 * i + 1]; in benchmark_small_model()328 small_model_gemms[i].depth = small_model_gemm_sizes[3 * i + 2]; in benchmark_small_model()329 small_model_gemms[i].cols = small_model_gemm_sizes[3 * i + 0]; in benchmark_small_model()