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

Function init_assembly_metadata

src/cpu/operators/CpuGemm.cpp:47–62  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

45namespace
46{
47cpu::AsmGemmInfo init_assembly_metadata(const GEMMInfo &info)
48{
49 cpu::AsmGemmInfo asm_info;
50 asm_info.method = cpu::AsmConvMethod::Im2Col;
51 asm_info.reinterpret_input_as_3d = info.reinterpret_input_as_3d();
52 asm_info.depth_output_gemm3d = info.depth_output_gemm3d();
53 asm_info.activation_info = info.activation_info();
54 asm_info.fast_mode = info.fast_math();
55 asm_info.fixed_format = info.fixed_format();
56 asm_info.weight_format = info.weight_format();
57 asm_info.accumulate = info.accumulate();
58 asm_info.transpose_b =
59 info.pretranspose_B(); // The "pretranspose_B" flag here is not the same as the pretranspose_B_array method. The flag here signals to pretranspose_B_array method if we want to perform additional transpose on B before the pretranspose_B_array method
60
61 return asm_info;
62}
63} // namespace
64
65void CpuGemm::configure(const ITensorInfo *a,

Callers 3

configureMethod · 0.70
validateMethod · 0.70
has_opt_implMethod · 0.70

Calls 8

accumulateMethod · 0.80
pretranspose_BMethod · 0.80
depth_output_gemm3dMethod · 0.45
activation_infoMethod · 0.45
fast_mathMethod · 0.45
fixed_formatMethod · 0.45
weight_formatMethod · 0.45

Tested by

no test coverage detected