Skip to content

Commit 0ef9dc6

Browse files
author
xuwei06
committed
Fix comment for norm_op
1 parent 6ecbf08 commit 0ef9dc6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/operators/norm_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ class NormOpMaker : public framework::OpProtoAndCheckerMaker {
3939
"M = C * H * W");
4040
AddComment(R"DOC(
4141
"Input shape: $(N, C, H, W)$
42-
Sclae shape: $(C, 1)$
42+
Scale shape: $(C, 1)$
4343
Output shape: $(N, C, H, W)$
4444
Where
4545
forward

paddle/operators/norm_op.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ class NormKernel : public framework::OpKernel<T> {
6666
context.GetPlace());
6767
auto tmp = framework::EigenVector<T, Eigen::RowMajor,
6868
Eigen::DenseIndex>::Flatten(tmp_tensor);
69-
// get colsum and sqrt , inverse
69+
// get colsum and sqrt , inverse
7070
auto dim = Eigen::array<int, 1>({{0}});
7171
tmp.device(*place) = x_square_batch_eigen.sum(dim);
7272
tmp.device(*place) = (tmp + epsilon).sqrt().inverse();

0 commit comments

Comments
 (0)