modify populate op parameter

pull/3847/head
yangruoqi713 5 years ago
parent 67005d42c8
commit c65d63a4af

File diff suppressed because it is too large Load Diff

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

Loading…
Cancel
Save