Searched refs:gemm_f_scratch (Results 1 – 2 of 2) sorted by relevance
283 scratch->AssureSize(meta::gemm_f_scratch(m, n, k, max_num_threads)); in MetaGemmFloat()288 scratch->AssureSize(meta::gemm_f_scratch(n, m, k, max_num_threads)); in MetaGemmFloat()
187 std::int32_t gemm_f_scratch(std::int32_t m, std::int32_t n, std::int32_t k, in gemm_f_scratch() function