未验证 提交 0bef76db 编写于 作者: M MaxwellDing 提交者: GitHub

refactor: merge MLU config API (#118)

上级 d81fbe8d
...@@ -238,13 +238,10 @@ void CxxConfig::set_mlu_core_number(int core_number) { ...@@ -238,13 +238,10 @@ void CxxConfig::set_mlu_core_number(int core_number) {
void CxxConfig::set_mlu_input_layout(DataLayoutType layout) { void CxxConfig::set_mlu_input_layout(DataLayoutType layout) {
mlu_input_layout_ = layout; mlu_input_layout_ = layout;
} }
void CxxConfig::set_mlu_use_first_conv(bool use_first_conv) { void CxxConfig::set_mlu_firstconv_param(const std::vector<float> &mean,
mlu_use_first_conv_ = use_first_conv; const std::vector<float> &std) {
} mlu_use_first_conv_ = true;
void CxxConfig::set_mlu_first_conv_mean(const std::vector<float> &mean) {
mlu_first_conv_mean_ = mean; mlu_first_conv_mean_ = mean;
}
void CxxConfig::set_mlu_first_conv_std(const std::vector<float> &std) {
mlu_first_conv_std_ = std; mlu_first_conv_std_ = std;
} }
lite_api::MLUCoreVersion CxxConfig::mlu_core_version() const { lite_api::MLUCoreVersion CxxConfig::mlu_core_version() const {
......
...@@ -184,17 +184,16 @@ class LITE_API CxxConfig : public ConfigBase { ...@@ -184,17 +184,16 @@ class LITE_API CxxConfig : public ConfigBase {
void set_mlu_core_version(lite_api::MLUCoreVersion core_version); void set_mlu_core_version(lite_api::MLUCoreVersion core_version);
// set MLU core number, which is used when compiling MLU kernels // set MLU core number, which is used when compiling MLU kernels
void set_mlu_core_number(int core_number); void set_mlu_core_number(int core_number);
// set MLU input layout. User can specify layout of input data to be NHWC,
// default is NCHW
void set_mlu_input_layout(DataLayoutType layout);
// whether use MLU's first conv kernel. First conv is a special kernel // whether use MLU's first conv kernel. First conv is a special kernel
// provided by MLU, its input is uint8, and also needs two 3-dimentional // provided by MLU, its input is uint8, and also needs two 3-dimentional
// vectors which save all inputs' mean and std values // vectors which save all inputs' mean and std values
void set_mlu_use_first_conv(bool use_first_conv); // set the 3-dimentional mean vector and 3-dimentional std vector used by
// set the 3-dimentional mean vector used by MLU's first conv // MLU's first conv
void set_mlu_first_conv_mean(const std::vector<float>& mean); void set_mlu_firstconv_param(const std::vector<float>& mean,
// set the 3-dimentional std vector used by MLU's first conv const std::vector<float>& std);
void set_mlu_first_conv_std(const std::vector<float>& std); // set MLU input layout. User can specify layout of input data to be NHWC,
// default is NCHW
void set_mlu_input_layout(DataLayoutType layout);
lite_api::MLUCoreVersion mlu_core_version() const; lite_api::MLUCoreVersion mlu_core_version() const;
int mlu_core_number() const; int mlu_core_number() const;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册