Lines Matching refs:gemms
70 double time_for_gemms(GemmContext* context, const std::vector<gemm_t>& gemms) { in time_for_gemms() argument
76 for (auto gemm : gemms) { in time_for_gemms()
86 std::vector<LhsType> lhs(pool_size * gemms.size()); in time_for_gemms()
87 std::vector<RhsType> rhs(pool_size * gemms.size()); in time_for_gemms()
88 std::vector<ResultType> result(pool_size * gemms.size()); in time_for_gemms()
91 for (std::size_t j = 0; j < gemms.size(); j++) { in time_for_gemms()
92 int k = i * gemms.size() + j; in time_for_gemms()
93 lhs[k].Resize(gemms[j].rows, gemms[j].depth); in time_for_gemms()
95 rhs[k].Resize(gemms[j].depth, gemms[j].cols); in time_for_gemms()
97 result[k].Resize(gemms[j].rows, gemms[j].cols); in time_for_gemms()
111 for (size_t j = 0; j < gemms.size(); j++) { in time_for_gemms()
112 int k = pool_index * gemms.size() + j; in time_for_gemms()
139 const std::vector<gemm_t>& gemms) { in gflops_for_gemms() argument
141 time_for_gemms<LhsType, RhsType, ResultType>(context, gemms); in gflops_for_gemms()
143 for (auto gemm : gemms) { in gflops_for_gemms()
213 const std::vector<gemm_t>& gemms, double mintime) { in benchmark_gemm_sizes() argument
229 time_for_gemms<LhsType, RhsType, ResultType>(context, gemms)); in benchmark_gemm_sizes()