21.02
|
#include "src/runtime/NEON/functions/NEGEMMAssemblyDispatch.h"
#include "arm_compute/runtime/NEON/NEScheduler.h"
#include "src/core/CPP/Validate.h"
#include "src/core/NEON/kernels/assembly/NEGEMMAssemblyWrapperKernel.h"
#include "src/core/NEON/kernels/assembly/arm_gemm.hpp"
#include <arm_neon.h>
#include <cstdlib>
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2021 Arm Limited. | |
unsigned int batches |
Definition at line 51 of file NEGEMMAssemblyDispatch.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 54 of file NEGEMMAssemblyDispatch.cpp.
Referenced by IndirectInputArg< T >::IndirectInputArg(), and IndirectOutputArg< T >::IndirectOutputArg().
unsigned int K |
Definition at line 50 of file NEGEMMAssemblyDispatch.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(), main(), and arm_compute::test::validation::matrix_multiply().
unsigned int M |
Definition at line 48 of file NEGEMMAssemblyDispatch.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(), main(), arm_compute::test::validation::matrix_multiply(), and arm_compute::test::validation::reference::safe_read().
unsigned int multis |
Definition at line 52 of file NEGEMMAssemblyDispatch.cpp.
unsigned int N |
Definition at line 49 of file NEGEMMAssemblyDispatch.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::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(), hysteresis(), NEMinMaxLocationKernel::is_parallelisable(), main(), arm_compute::test::validation::matrix_multiply(), arm_compute::test::validation::reference::safe_read(), NEFFTDigitReverseKernel::validate(), NEFFT1D::validate(), and CLFFT1D::validate().
unsigned int sections |
Definition at line 53 of file NEGEMMAssemblyDispatch.cpp.