提交 ecb87385 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3847 modify populate op parameter

Merge pull request !3847 from yangruoqi713/lite
...@@ -22,7 +22,20 @@ ...@@ -22,7 +22,20 @@
#include "src/runtime/kernel/arm/opclib/op_base.h" #include "src/runtime/kernel/arm/opclib/op_base.h"
namespace mindspore::kernel { namespace mindspore::kernel {
typedef OpParameter *(*PopulateParameterFunc)(const lite::Primitive *);
class PopulateParameterRegistry {
public:
PopulateParameterRegistry();
~PopulateParameterRegistry() = default;
static PopulateParameterRegistry *GetInstance();
PopulateParameterFunc GetParameterFunc(const schema::PrimitiveType &type);
protected:
PopulateParameterFunc populate_parameter_funcs_[schema::PrimitiveType_MAX + 1];
};
OpParameter *PopulateParameter(const lite::Primitive *primitive); OpParameter *PopulateParameter(const lite::Primitive *primitive);
} // namespace mindspore::kernel } // namespace mindspore::kernel
#endif // MINDSPORE_LITE_SRC_POPULATE_PARAMETER_H_ #endif // MINDSPORE_LITE_SRC_POPULATE_PARAMETER_H_
...@@ -24,7 +24,7 @@ ...@@ -24,7 +24,7 @@
#include "src/runtime/kernel/arm/opclib/arithmetic_common.h" #include "src/runtime/kernel/arm/opclib/arithmetic_common.h"
struct ArithmeticParameter { struct ArithmeticParameter {
OpParameter op_parameter; OpParameter op_parameter_;
bool broadcasting_; bool broadcasting_;
size_t ndim_; size_t ndim_;
int in_shape0_[5]; int in_shape0_[5];
......
...@@ -20,7 +20,7 @@ ...@@ -20,7 +20,7 @@
#include "src/runtime/kernel/arm/opclib/op_base.h" #include "src/runtime/kernel/arm/opclib/op_base.h"
struct SoftmaxParameter { struct SoftmaxParameter {
OpParameter op_parameter; OpParameter op_parameter_;
int32_t axis_; int32_t axis_;
int element_size_; int element_size_;
int n_dim_; int n_dim_;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册