Searched defs:gemm_3d_depth (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuGemmConv2d.cpp | 102 …bool enable_fast_math, int gemm_3d_depth, bool fixed_format, arm_compute::WeightFormat weight_form… in configure_mm()
182 …const ActivationLayerInfo &act_info, bool enable_fast_math, int gemm_3d_depth, bool skip_im2col, b… in validate_mm()
242 …nst ITensorInfo *weights_info, const ActivationLayerInfo &act_info, int gemm_3d_depth, bool skip_i… in validate_gemm3d()
362 const unsigned int gemm_3d_depth = _skip_col2im ? conv_h : 0; in configure() local
414 const unsigned int gemm_3d_depth = skip_col2im ? conv_h : 0; in has_opt_impl() local
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ |
| H A D | ClGemmConv2d.cpp | 64 …int gemm_3d_depth, const ActivationLayerInfo &act_info, const experimental::PostOpList<ITensorInfo… in configure_mm()
121 …const GEMMLowpOutputStageInfo &gemmlowp_output_stage, int gemm_3d_depth, bool skip_im2col, const A… in validate_mm()
319 const unsigned int gemm_3d_depth = (data_layout == DataLayout::NHWC) ? conv_h : 0; in configure() local
521 const unsigned int gemm_3d_depth = (data_layout == DataLayout::NHWC) ? conv_h : 0; in validate() local
|