21.08
|
#include "src/runtime/cpu/operators/internal/CpuGemmAssemblyDispatch.h"
#include "arm_compute/runtime/NEON/NEScheduler.h"
#include "src/core/CPP/Validate.h"
#include "src/core/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h"
#include "src/core/cpu/kernels/assembly/arm_gemm.hpp"
#include "src/core/helpers/MemoryHelpers.h"
#include "src/core/utils/AssemblyUtils.h"
#include "src/runtime/cpu/utils/CpuAuxTensorHandler.h"
#include <arm_neon.h>
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2021 Arm Limited. | |
arm_compute::cpu | |
unsigned int batches |
Definition at line 57 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_conv::depthwise::common::depthwise_multiplier_execute(), PoolingDepthfirstGeneric< strategy >::execute(), PoolingDepthfirstGenericQuantized< strategy >::execute(), DepthwiseDepthfirst< strategy >::execute(), PoolingDepthfirst< strategy >::execute(), DepthwiseDepthfirstWithMultiplierQuantized< strategy >::execute(), DepthwiseDepthfirstQuantized< strategy >::execute(), DepthwiseDepthfirstWithMultiplier< strategy >::execute(), 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 60 of file CpuGemmAssemblyDispatch.cpp.
Referenced by IndirectInputArg< T >::IndirectInputArg(), and IndirectOutputArg< T >::IndirectOutputArg().
unsigned int K |
Definition at line 56 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(), and arm_compute::test::validation::matrix_multiply().
unsigned int M |
Definition at line 54 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(), and arm_compute::test::validation::reference::safe_read().
unsigned int multis |
Definition at line 58 of file CpuGemmAssemblyDispatch.cpp.
unsigned int N |
Definition at line 55 of file CpuGemmAssemblyDispatch.cpp.
Referenced by arm_compute::test::validation::reference::channel_shuffle(), NEFFT1D::configure(), CLFFT1D::configure(), arm_compute::test::validation::DATA_TEST_CASE(), 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(), NDCoordinate< N >::get_position(), main(), arm_compute::test::validation::matrix_multiply(), arm_compute::test::validation::reference::safe_read(), NDCoordinate< N >::set_position(), NEFFTDigitReverseKernel::validate(), NEFFT1D::validate(), and CLFFT1D::validate().
unsigned int sections |
Definition at line 59 of file CpuGemmAssemblyDispatch.cpp.
Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize >::execute().