Searched defs:gemm_kernel (Results 1 – 3 of 3) sorted by relevance
73 auto gemm_kernel = auto_heuristics::select_mlgo_gemm_kernel(query, reshape_b_only_on_first_run); in auto_select_gemm_kernel() local74 if(bool(gemm_kernel)) in auto_select_gemm_kernel() local
78 auto gemm_kernel = auto_heuristics::select_mlgo_gemm_kernel(query, reshape_b_only_on_first_run); in auto_select_gemm_kernel() local79 if(bool(gemm_kernel)) in auto_select_gemm_kernel() local
122 MkldnnGemmKernel gemm_kernel; in TEST() local