OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

!3847 modify populate op parameter 

Merge pull request !3847 from yangruoqi713/lite
This commit is contained in:
mindspore-ci-bot 2020-08-03 10:14:09 +08:00 committed by Gitee
parent 49ba473bcc c65d63a4af
commit ecb8738524
4 changed files with 653 additions and 630 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1264
mindspore/lite/src/populate_parameter.cc
View File

File diff suppressed because it is too large Load Diff

15
mindspore/lite/src/populate_parameter.h
View File

@ -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_

2
mindspore/lite/src/runtime/kernel/arm/opclib/arithmetic_common.h
View File

@ -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];

2
mindspore/lite/src/runtime/kernel/arm/opclib/fp32/softmax.h
View File

@ -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_;