MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / get_wt_method

Method get_wt_method

src/cpu/operators/CpuGemmConv2d.cpp:162–171  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

160} // namespace
161
162CpuGemmConv2d::WeightTransformMethod CpuGemmConv2d::get_wt_method(const ITensorInfo &weights)
163{
164 // TODO: Extend ReinterpretThenTranspose support for quantized data types COMPMID-6596
165 if (is_data_type_quantized(weights.data_type()))
166 {
167 return WeightTransformMethod::FusedReshapeAndTranspose;
168 }
169 return has_holes(weights) ? WeightTransformMethod::ReshapeThenTranspose
170 : WeightTransformMethod::ReinterpretThenTranspose;
171}
172
173CpuGemmConv2d::SkipInfo CpuGemmConv2d::skip_im_col_info(const ITensorInfo *src,
174 const ITensorInfo *weights,

Callers

nothing calls this directly

Calls 3

is_data_type_quantizedFunction · 0.85
has_holesFunction · 0.85
data_typeMethod · 0.45

Tested by

no test coverage detected