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

!3847 modify populate op parameter

Merge pull request !3847 from yangruoqi713/lite
......@@ -22,7 +22,20 @@
#include "src/runtime/kernel/arm/opclib/op_base.h"
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);
} // namespace mindspore::kernel
#endif // MINDSPORE_LITE_SRC_POPULATE_PARAMETER_H_
......@@ -24,7 +24,7 @@
#include "src/runtime/kernel/arm/opclib/arithmetic_common.h"
struct ArithmeticParameter {
OpParameter op_parameter;
OpParameter op_parameter_;
bool broadcasting_;
size_t ndim_;
int in_shape0_[5];
......
......@@ -20,7 +20,7 @@
#include "src/runtime/kernel/arm/opclib/op_base.h"
struct SoftmaxParameter {
OpParameter op_parameter;
OpParameter op_parameter_;
int32_t axis_;
int element_size_;
int n_dim_;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册