提交 b0c81e3e 编写于 作者: Y Yao,kun

Code style

上级 d9a7ecd4
...@@ -35,7 +35,6 @@ void DropoutKernel<CPU, float>::Compute(const DropoutParam &param) const { ...@@ -35,7 +35,6 @@ void DropoutKernel<CPU, float>::Compute(const DropoutParam &param) const {
DropoutFunctor<float> func_; DropoutFunctor<float> func_;
math::Transform trans; math::Transform trans;
trans(input_x_ptr, input_x_ptr + input_x->numel(), out_ptr, func_); trans(input_x_ptr, input_x_ptr + input_x->numel(), out_ptr, func_);
} }
} // namespace operators } // namespace operators
} // namespace paddle_mobile } // namespace paddle_mobile
...@@ -877,22 +877,22 @@ class Im2SequenceParam : public OpParam { ...@@ -877,22 +877,22 @@ class Im2SequenceParam : public OpParam {
vector<int> strides_; vector<int> strides_;
vector<int> paddings_; vector<int> paddings_;
}; };
class DropoutParam : public OpParam { class DropoutParam : public OpParam {
public: public:
DropoutParam(const VariableNameMap &inputs, const VariableNameMap &outputs, DropoutParam(const VariableNameMap &inputs, const VariableNameMap &outputs,
const AttributeMap &attrs, const Scope &scope) { const AttributeMap &attrs, const Scope &scope) {
input_x_ = InputXFrom<LoDTensor>(inputs, scope); input_x_ = InputXFrom<LoDTensor>(inputs, scope);
out_ = OutFrom<LoDTensor>(outputs, scope); out_ = OutFrom<LoDTensor>(outputs, scope);
} }
const Tensor *InputX() const { return input_x_; } const Tensor *InputX() const { return input_x_; }
Tensor *Out() const { return out_; } Tensor *Out() const { return out_; }
private: private:
Tensor *input_x_; Tensor *input_x_;
Tensor *out_; Tensor *out_;
}; };
} // namespace operators } // namespace operators
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册