Searched refs:_skip_im2col (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuGemmConv2d.cpp | 96 …_data_layout(DataLayout::NCHW), _skip_im2col(false), _skip_col2im(false), _is_quantized(false), _i… in CpuGemmConv2d()
105 …src, weights, biases, dst, act_info, enable_fast_math, gemm_3d_depth, _skip_im2col, fixed_format, … in configure_mm()
109 … gemm_3d_depth, _skip_im2col /* Reinterpret the input as 3D if im2col is skipped */, in configure_mm()
159 …tmp_weights, biases, dst, GEMMInfo(false, false, true, gemm_3d_depth, _skip_im2col, false, output_… in configure_mm()
285 …_skip_im2col = (data_layout == DataLayout::NHWC && kernel_width == 1 && kernel_height == 1 && con… in configure()
305 _skip_im2col = skip_info.skip_im2col; in configure()
322 if(!_skip_im2col) in configure()
562 if(!_skip_im2col) in run()
583 if(_skip_im2col) in run()
|
| D | CpuGemmConv2d.h | 233 bool _skip_im2col; variable
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemmConv2d.cpp | 57 …_gemm_output(), _skip_im2col(false), _skip_col2im(false), _is_quantized(false), _fuse_activation(t… in ClGemmConv2d()
67 …idate_mm(src, weights, biases, dst, gemmlowp_output_stage, gemm_3d_depth, _skip_im2col, act_info)); in configure_mm()
73 _skip_im2col, // reinterpret_input_as_3d in configure_mm()
187 …_skip_im2col = (data_layout == DataLayout::NHWC && kernel_width == 1 && kernel_height == 1 && con… in configure()
232 if(!_skip_im2col) in configure()
557 if(!_skip_im2col) in run()
|
| D | ClGemmConv2d.h | 175 bool _skip_im2col; variable
|