|
|
|
@ -19,56 +19,56 @@
|
|
|
|
|
#include "src/ops/ops.h"
|
|
|
|
|
#include "utils/log_adapter.h"
|
|
|
|
|
#include "schema/ops_generated.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/op_base.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/arg_min_max.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/cast.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/concat_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/slice.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/broadcast_to.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/reshape_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/stack.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/unstack.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/depth_to_space.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/conv_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/pooling.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/matmul.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/softmax_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/tile.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/topk.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/reduce.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/activation.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/arithmetic.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fused_batchnorm.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/batchnorm.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/power.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/range.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/local_response_norm.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/expandDims.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/arithmetic_self_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/pad_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/fill.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/transpose.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/split_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/squeeze.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/gather.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/reverse.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/reverse_sequence.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/unique.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/scale.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/gatherNd.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/resize.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/scatter_nd.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/batch_to_space.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/crop.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/op_base.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/arg_min_max.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/cast.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/concat_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/slice.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/broadcast_to.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/reshape_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/stack.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/unstack.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/depth_to_space.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/conv_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/pooling.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/matmul.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/softmax_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/tile.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/topk.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/reduce.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/activation.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/arithmetic.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fused_batchnorm.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/batchnorm.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/power.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/range.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/local_response_norm.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/expandDims.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/arithmetic_self_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/pad_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/fill.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/transpose.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/split_parameter.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/squeeze.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/gather.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/reverse.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/reverse_sequence.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/unique.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/scale.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/gatherNd.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/resize.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/scatter_nd.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/batch_to_space.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/crop.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/fp32/flatten.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/unsqueeze.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/one_hot.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/strided_slice.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/unsqueeze.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/one_hot.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/strided_slice.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/base/prior_box.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/space_to_depth.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/space_to_batch.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/int8/quant_dtype_cast.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/opclib/fp32/lstm.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/space_to_depth.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/space_to_batch.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/int8/quant_dtype_cast.h"
|
|
|
|
|
#include "src/runtime/kernel/arm/nnacl/fp32/lstm.h"
|
|
|
|
|
|
|
|
|
|
namespace mindspore::kernel {
|
|
|
|
|
OpParameter *PopulateBatchNorm(const lite::Primitive *primitive) {
|
|
|
|
|