| 45 | namespace |
| 46 | { |
| 47 | cpu::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 | |
| 65 | void CpuGemm::configure(const ITensorInfo *a, |
no test coverage detected