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

* Also fix unit test
cblas_new
Yu Yang 8 years ago
parent 1ed237c11e
commit 5847b96a61

@ -278,7 +278,7 @@ std::ostream& operator<<(std::ostream& os, const DDim& ddim) {
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) {
*this = make_ddim(init_list);

@ -1,5 +1,5 @@
#include <paddle/framework/op_registry.h>
#include <paddle/operators/sigmoid_op.h>
#include <paddle/operators/softmax_op.h>
REGISTER_OP_GPU_KERNEL(
softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace>);

Loading…
Cancel
Save