|
|
|
@@ -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) { |
|
|
|
|