Lines Matching refs:gemms
62 double time_for_gemms(GemmContext* context, const std::vector<gemm_t>& gemms) { in time_for_gemms() argument
68 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()
103 for (size_t j = 0; j < gemms.size(); j++) { in time_for_gemms()
104 size_t k = pool_index * gemms.size() + j; in time_for_gemms()
131 const std::vector<gemm_t>& gemms) { in gflops_for_gemms() argument
133 time_for_gemms<LhsType, RhsType, ResultType>(context, gemms); in gflops_for_gemms()
135 for (auto gemm : gemms) { in gflops_for_gemms()
205 const std::vector<gemm_t>& gemms, double mintime) { in benchmark_gemm_sizes() argument
221 time_for_gemms<LhsType, RhsType, ResultType>(context, gemms)); in benchmark_gemm_sizes()