CpuGemmAssemblyDispatch.cpp File Reference

#include "src/cpu/operators/internal/CpuGemmAssemblyDispatch.h"
#include "arm_compute/runtime/NEON/NEScheduler.h"
#include "src/core/CPP/Validate.h"
#include "src/core/NEON/kernels/arm_gemm/utils.hpp"
#include "src/core/helpers/MemoryHelpers.h"
#include "src/core/utils/AssemblyUtils.h"
#include "src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h"
#include "src/cpu/kernels/assembly/arm_gemm.hpp"
#include "src/cpu/utils/CpuAuxTensorHandler.h"
#include <arm_neon.h>

Go to the source code of this file.

Namespaces
	arm_compute
	Copyright (c) 2017-2023 Arm Limited.
	arm_compute::cpu

Variable Documentation

batches

unsigned int batches

Definition at line 98 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(), PoolingDepthfirstGenericQuantized< strategy >::execute(), DepthwisePlanar< TInput, TWeight, TOutput, TAccum, OutputStage >::get_working_size(), arm_compute::test::validation::reference::im2col_nchw(), arm_compute::test::validation::reference::im2col_nhwc(), and arm_compute::test::validation::reference::reverse().

indirect

bool indirect

Definition at line 101 of file CpuGemmAssemblyDispatch.cpp.

Referenced by IndirectInputArg< T >::IndirectInputArg(), IndirectOutputArg< T >::IndirectOutputArg(), and DepthwiseDepthfirst< TInput, TWeight, TOutput, TAccum, OutputStage >::initialise_working_space().

K

unsigned int K

Definition at line 97 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().

M

unsigned int M

Definition at line 95 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().

multis

unsigned int multis

Definition at line 99 of file CpuGemmAssemblyDispatch.cpp.

N

unsigned int N

Definition at line 96 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(), arm_compute::get_layout_map(), 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().

sections

unsigned int sections

Definition at line 100 of file CpuGemmAssemblyDispatch.cpp.

Referenced by GemmHybridIndirect< strategy, To, Tr, OutputStage, SeparateQuantize, FixedFormat >::execute().