Searched refs:gemms (Results 1 – 1 of 1) sorted by relevance
75 double time_for_gemms(GemmContext* context, const std::vector<gemm_t>& gemms) { in time_for_gemms() argument81 for (auto gemm : gemms) { in time_for_gemms()91 std::vector<LhsType> lhs(pool_size * gemms.size()); in time_for_gemms()92 std::vector<RhsType> rhs(pool_size * gemms.size()); in time_for_gemms()93 std::vector<ResultType> result(pool_size * gemms.size()); in time_for_gemms()96 for (std::size_t j = 0; j < gemms.size(); j++) { in time_for_gemms()97 int k = i * gemms.size() + j; in time_for_gemms()98 lhs[k].Resize(gemms[j].rows, gemms[j].depth); in time_for_gemms()100 rhs[k].Resize(gemms[j].depth, gemms[j].cols); in time_for_gemms()102 result[k].Resize(gemms[j].rows, gemms[j].cols); in time_for_gemms()[all …]