Browse Source

modify populate op parameter

tags/v0.7.0-beta
yangruoqi713 5 years ago
parent
commit
c65d63a4af
4 changed files with 654 additions and 631 deletions
  1. +638
    -628
      mindspore/lite/src/populate_parameter.cc
  2. +14
    -1
      mindspore/lite/src/populate_parameter.h
  3. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/opclib/arithmetic_common.h
  4. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/opclib/fp32/softmax.h

+ 638
- 628
mindspore/lite/src/populate_parameter.cc
File diff suppressed because it is too large
View File


+ 14
- 1
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_


+ 1
- 1
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];


+ 1
- 1
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_;


Loading…
Cancel
Save