Searched defs:gemms (Results 1 – 3 of 3) sorted by relevance
71 double time_for_gemms(GemmContext* context, const std::vector<gemm_t>& gemms) { in time_for_gemms()140 const std::vector<gemm_t>& gemms) { in gflops_for_gemms()214 const std::vector<gemm_t>& gemms, double mintime) { in benchmark_gemm_sizes()
237 auto gemms = gemm_implementation_list<Top, Tret, OutputStage>(); in find_implementation() local293 auto gemms = gemm_implementation_list<Top, Tret, OutputStage>(); in get_compatible_kernels() local
153 std::vector<HloDotInstruction*> gemms; in GetRelevantDots() local