23.11
|
#include "src/core/NEON/kernels/NEFuseBatchNormalizationKernel.h"
#include "arm_compute/core/Helpers.h"
#include "arm_compute/core/ITensor.h"
#include "arm_compute/core/TensorInfo.h"
#include "arm_compute/core/Utils.h"
#include "arm_compute/core/Validate.h"
#include "arm_compute/core/Window.h"
#include "src/common/cpuinfo/CpuIsaInfo.h"
#include "src/core/common/Registrars.h"
#include "src/core/CPP/Validate.h"
#include "src/core/helpers/AutoConfiguration.h"
#include "src/core/helpers/WindowHelpers.h"
#include "src/core/NEON/wrapper/wrapper.h"
#include "src/cpu/kernels/fuse_batch_normalization/list.h"
#include <map>
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2023 Arm Limited. | |
DataLayout dl |
Definition at line 50 of file NEFuseBatchNormalizationKernel.cpp.
Referenced by arm_compute::error_on_data_layout_not_in(), arm_compute::string_from_data_layout(), and arm_compute::experimental::dynamic_fusion::to_ckw().
DataType dt |
Definition at line 49 of file NEFuseBatchNormalizationKernel.cpp.
FuseBatchNormalizationType fbn_type |
Definition at line 51 of file NEFuseBatchNormalizationKernel.cpp.
Referenced by CLFuseBatchNormalizationKernel::configure(), NEFuseBatchNormalizationKernel::configure(), NEFuseBatchNormalization::configure(), CLFuseBatchNormalization::configure(), NEFuseBatchNormalizationKernel::validate(), NEFuseBatchNormalization::validate(), CLFuseBatchNormalizationKernel::validate(), and CLFuseBatchNormalization::validate().
const FBNSelectorPtr is_selected |
Definition at line 70 of file NEFuseBatchNormalizationKernel.cpp.
cpuinfo::CpuIsaInfo isa |
Definition at line 52 of file NEFuseBatchNormalizationKernel.cpp.
Referenced by arm_compute::cpuinfo::init_cpu_isa_from_hwcaps(), and arm_compute::cpuinfo::init_cpu_isa_from_regs().
const char* name |
Definition at line 69 of file NEFuseBatchNormalizationKernel.cpp.
FBNUKernelPtr ukernel |
Definition at line 71 of file NEFuseBatchNormalizationKernel.cpp.