Home
last modified time | relevance | path

Searched refs:mm_res_info (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLQLSTMLayer.cpp47 … float gemmlowp_scale, const TensorInfo *mm_res_info, const TensorInfo *outstage_tensor_info) in validate_mm() argument
49 …ETURN_ON_ERROR(CLGEMMLowpMatrixMultiplyCore::validate(mm_input, mm_weights, nullptr, mm_res_info)); in validate_mm()
51 …ARM_COMPUTE_RETURN_ON_ERROR(CLGEMMLowpOutputStage::validate(mm_res_info, bias, outstage_tensor_inf… in validate_mm()
142 … const TensorInfo &mm_res_info, const TensorInfo &outstage_tensor_info) in configure_mm() argument
147 mm_res->allocator()->init(mm_res_info); in configure_mm()
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEQLSTMLayer.cpp45 … float gemmlowp_scale, const TensorInfo *mm_res_info, const TensorInfo *outstage_tensor_info) in validate_mm() argument
47 …ETURN_ON_ERROR(NEGEMMLowpMatrixMultiplyCore::validate(mm_input, mm_weights, nullptr, mm_res_info)); in validate_mm()
49 …ARM_COMPUTE_RETURN_ON_ERROR(NEGEMMLowpOutputStage::validate(mm_res_info, bias, outstage_tensor_inf… in validate_mm()
197 … const TensorInfo &mm_res_info, const TensorInfo &outstage_tensor_info) in configure_mm() argument
202 mm_res->allocator()->init(mm_res_info); in configure_mm()
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/
H A DCLQLSTMLayer.h272 const TensorInfo &mm_res_info, const TensorInfo &outstage_tensor_info);
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
H A DNEQLSTMLayer.h224 const TensorInfo &mm_res_info, const TensorInfo &outstage_tensor_info);