提交 5847b96a 编写于 作者: Y Yu Yang

Follow comments, correct implement `DDim::size()`

* Also fix unit test
上级 1ed237c1
...@@ -278,7 +278,7 @@ std::ostream& operator<<(std::ostream& os, const DDim& ddim) { ...@@ -278,7 +278,7 @@ std::ostream& operator<<(std::ostream& os, const DDim& ddim) {
return os; return os;
} }
ssize_t DDim::size() const { return product(*this); } ssize_t DDim::size() const { return arity(*this); }
DDim::DDim(std::initializer_list<int> init_list) { DDim::DDim(std::initializer_list<int> init_list) {
*this = make_ddim(init_list); *this = make_ddim(init_list);
......
#include <paddle/framework/op_registry.h> #include <paddle/framework/op_registry.h>
#include <paddle/operators/sigmoid_op.h> #include <paddle/operators/softmax_op.h>
REGISTER_OP_GPU_KERNEL( REGISTER_OP_GPU_KERNEL(
softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace>); softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace>);
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册