Searched refs:gemms (Results 1 – 2 of 2) sorted by relevance
62 double time_for_gemms(GemmContext* context, const std::vector<gemm_t>& gemms) { in time_for_gemms() argument68 for (auto gemm : gemms) { in time_for_gemms()78 std::vector<LhsType> lhs(pool_size * gemms.size()); in time_for_gemms()79 std::vector<RhsType> rhs(pool_size * gemms.size()); in time_for_gemms()80 std::vector<ResultType> result(pool_size * gemms.size()); in time_for_gemms()83 for (std::size_t j = 0; j < gemms.size(); j++) { in time_for_gemms()84 int k = i * gemms.size() + j; in time_for_gemms()85 lhs[k].Resize(gemms[j].rows, gemms[j].depth); in time_for_gemms()87 rhs[k].Resize(gemms[j].depth, gemms[j].cols); in time_for_gemms()89 result[k].Resize(gemms[j].rows, gemms[j].cols); in time_for_gemms()[all …]
78171 gemms %15293