提交 ff5d68a9 编写于 作者: Z zhangyang

change class name

上级 157e2fcd
...@@ -439,7 +439,7 @@ class PoolParam : public OpParam { ...@@ -439,7 +439,7 @@ class PoolParam : public OpParam {
bool isCeilMode() const { return ceil_mode_; } bool isCeilMode() const { return ceil_mode_; }
bool isGlobalPooling() const { return gloabal_pooling_; } bool isGlobalPooling() const { return global_pooling_; }
private: private:
Tensor *input_; Tensor *input_;
...@@ -454,9 +454,9 @@ class PoolParam : public OpParam { ...@@ -454,9 +454,9 @@ class PoolParam : public OpParam {
#endif #endif
#ifdef FUSION_POOLBN_OP #ifdef FUSION_POOLBN_OP
class PoolBNParam : OpParam { class FusionPoolBNParam : OpParam {
public: public:
PoolBNParam(const VariableNameMap &inputs, const VariableNameMap &outputs, FusionPoolBNParam(const VariableNameMap &inputs, const VariableNameMap &outputs,
const AttributeMap &attrs, const Scope &scope) { const AttributeMap &attrs, const Scope &scope) {
input_ = InputXFrom<LoDTensor>(inputs, scope); input_ = InputXFrom<LoDTensor>(inputs, scope);
pooling_type_ = GetAttr<string>("pooling_type", attrs); pooling_type_ = GetAttr<string>("pooling_type", attrs);
...@@ -486,7 +486,7 @@ class PoolBNParam : OpParam { ...@@ -486,7 +486,7 @@ class PoolBNParam : OpParam {
bool isCeilMode() const { return ceil_mode_; } bool isCeilMode() const { return ceil_mode_; }
bool isGlobalPooling() const { return gloabal_pooling_; } bool isGlobalPooling() const { return global_pooling_; }
Tensor *OutputY() const { return output_y_; } Tensor *OutputY() const { return output_y_; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册