24.02.1
|
#include "src/cpu/operators/internal/CpuGemmAssemblyDispatch.h"
#include "arm_compute/runtime/NEON/NEScheduler.h"
#include "src/core/CPP/Validate.h"
#include "src/core/helpers/MemoryHelpers.h"
#include "src/core/NEON/kernels/arm_gemm/utils.hpp"
#include "src/core/utils/AssemblyUtils.h"
#include "src/cpu/kernels/assembly/arm_gemm.hpp"
#include "src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h"
#include "src/cpu/operators/CpuTranspose.h"
#include "src/cpu/utils/CpuAuxTensorHandler.h"
#include <arm_neon.h>
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2024 Arm Limited. | |
arm_compute::cpu | |
unsigned int batches |
Definition at line 105 of file CpuGemmAssemblyDispatch.cpp.
Referenced by arm_compute::test::validation::reference::channel_shuffle(), arm_compute::test::validation::reference::col2im(), FullyConnectedLayerNode::compute_output_descriptor(), arm_compute::test::validation::reference::depthconcatenate_layer(), arm_compute::test::validation::reference::im2col_nchw(), arm_compute::test::validation::reference::im2col_nhwc(), and arm_compute::test::validation::reference::reverse().
bool indirect |
Definition at line 108 of file CpuGemmAssemblyDispatch.cpp.
unsigned int K |
Definition at line 104 of file CpuGemmAssemblyDispatch.cpp.
Referenced by arm_compute::test::validation::reference::gemm(), arm_compute::test::validation::reference::gemm_mixed_precision(), arm_compute::test::validation::reference::gemm_reshape_lhs_matrix(), arm_compute::test::validation::reference::gemm_reshape_rhs_matrix(), arm_compute::test::validation::reference::gemmlowp_matrix_multiply_core(), main(), arm_compute::test::validation::matrix_multiply(), and arm_compute::test::validation::suggest_mac_dst_q_info_and_bias().
unsigned int M |
Definition at line 102 of file CpuGemmAssemblyDispatch.cpp.
Referenced by arm_compute::test::validation::reference::channel_shuffle(), arm_compute::misc::shape_calculator::compute_interleaved_shape(), arm_compute::test::validation::reference::gemm(), arm_compute::test::validation::reference::gemm_interleave_blocked(), arm_compute::test::validation::reference::gemm_mixed_precision(), arm_compute::test::validation::reference::gemm_reshape_lhs_matrix(), arm_compute::test::validation::reference::gemmlowp_matrix_multiply_core(), main(), arm_compute::test::validation::matrix_multiply(), NEFFTRadixStageKernel::run(), and arm_compute::test::validation::reference::safe_read().
unsigned int multis |
Definition at line 106 of file CpuGemmAssemblyDispatch.cpp.
unsigned int N |
Definition at line 103 of file CpuGemmAssemblyDispatch.cpp.
Referenced by arm_compute::test::validation::reference::channel_shuffle(), NEFFT1D::configure(), CLFFT1D::configure(), arm_compute::helpers::fft::decompose_stages(), arm_compute::test::validation::reference::dequantization_layer(), arm_compute::helpers::fft::digit_reverse_indices(), arm_compute::test::validation::reference::gemm(), arm_compute::test::validation::reference::gemm_interleave_blocked(), arm_compute::test::validation::reference::gemm_mixed_precision(), arm_compute::test::validation::reference::gemm_reshape_rhs_matrix(), arm_compute::test::validation::reference::gemmlowp_matrix_multiply_core(), arm_compute::get_layout_map(), NDCoordinate< N >::get_position(), main(), arm_compute::test::validation::matrix_multiply(), NEFFTRadixStageKernel::run(), arm_compute::test::validation::reference::safe_read(), NDCoordinate< N >::set_position(), NEFFT1D::validate(), and CLFFT1D::validate().
unsigned int sections |
Definition at line 107 of file CpuGemmAssemblyDispatch.cpp.
Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute().