Home
last modified time | relevance | path

Searched refs:gemm_kernel (Results 1 – 3 of 3) sorted by relevance

/external/ComputeLibrary/src/gpu/cl/operators/
DClGemm.cpp78 auto gemm_kernel = auto_heuristics::select_mlgo_gemm_kernel(query, reshape_b_only_on_first_run); in auto_select_gemm_kernel() local
79 if(bool(gemm_kernel)) in auto_select_gemm_kernel() local
81 if(validate_gemm_kernel(gemm_kernel.gemm_type)) in auto_select_gemm_kernel()
83 …ITH_FORMAT_CORE("Use gemm kernel from mlgo heuristics: %s.", to_string(gemm_kernel.gemm_type).c_st… in auto_select_gemm_kernel()
84 return gemm_kernel.gemm_type; in auto_select_gemm_kernel()
87 gemm_kernel = auto_heuristics::select_default_gemm_kernel(query, reshape_b_only_on_first_run); in auto_select_gemm_kernel()
88 …_FORMAT_CORE("Use gemm kernel from default heuristics: %s.", to_string(gemm_kernel.gemm_type).c_st… in auto_select_gemm_kernel()
89 return gemm_kernel.gemm_type; in auto_select_gemm_kernel()
DClGemmLowpMatrixMultiplyCore.cpp73 auto gemm_kernel = auto_heuristics::select_mlgo_gemm_kernel(query, reshape_b_only_on_first_run); in auto_select_gemm_kernel() local
74 if(bool(gemm_kernel)) in auto_select_gemm_kernel() local
76 if(validate_gemm_kernel(gemm_kernel.gemm_type)) in auto_select_gemm_kernel()
78 …ITH_FORMAT_CORE("Use gemm kernel from mlgo heuristics: %s.", to_string(gemm_kernel.gemm_type).c_st… in auto_select_gemm_kernel()
79 return gemm_kernel.gemm_type; in auto_select_gemm_kernel()
82 gemm_kernel = auto_heuristics::select_default_gemm_kernel(query, reshape_b_only_on_first_run); in auto_select_gemm_kernel()
83 …_FORMAT_CORE("Use gemm kernel from default heuristics: %s.", to_string(gemm_kernel.gemm_type).c_st… in auto_select_gemm_kernel()
84 return gemm_kernel.gemm_type; in auto_select_gemm_kernel()
/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc122 MkldnnGemmKernel gemm_kernel; in TEST() local
123 gemm_kernel(output_mapper, lhs.data(), rhs.data(), m, k, n, /*alpha=*/1.0, in TEST()