Searched defs:skip_im2col (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuGemmConv2d.cpp | 71 …const bool skip_im2col = (data_layout == DataLayout::NHWC && kernel_width == 1 && kernel_height ==… in skip_im_col_info() local
182 …ationLayerInfo &act_info, bool enable_fast_math, int gemm_3d_depth, bool skip_im2col, bool fixed_f… in validate_mm()
242 …TensorInfo *weights_info, const ActivationLayerInfo &act_info, int gemm_3d_depth, bool skip_im2col) in validate_gemm3d()
412 const bool skip_im2col = skip_info.skip_im2col; in has_opt_impl() local
474 const bool skip_im2col = skip_info.skip_im2col, skip_col2im = skip_info.skip_col2im; in validate() local
|
| D | CpuGemmConv2d.h | 188 bool skip_im2col; member
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemmConv2d.cpp | 121 … GEMMLowpOutputStageInfo &gemmlowp_output_stage, int gemm_3d_depth, bool skip_im2col, const Activa… in validate_mm()
384 …const bool skip_im2col = (data_layout == DataLayout::NHWC && kernel_width == 1 && k… in validate() local
|