File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -39,7 +39,7 @@ class NormOpMaker : public framework::OpProtoAndCheckerMaker {
39
39
" M = C * H * W" );
40
40
AddComment (R"DOC(
41
41
"Input shape: $(N, C, H, W)$
42
- Sclae shape: $(C, 1)$
42
+ Scale shape: $(C, 1)$
43
43
Output shape: $(N, C, H, W)$
44
44
Where
45
45
forward
Original file line number Diff line number Diff line change @@ -66,7 +66,7 @@ class NormKernel : public framework::OpKernel<T> {
66
66
context.GetPlace ());
67
67
auto tmp = framework::EigenVector<T, Eigen::RowMajor,
68
68
Eigen::DenseIndex>::Flatten (tmp_tensor);
69
- // get colsum and sqrt , inverse
69
+ // get colsum and sqrt , inverse
70
70
auto dim = Eigen::array<int , 1 >({{0 }});
71
71
tmp.device (*place) = x_square_batch_eigen.sum (dim);
72
72
tmp.device (*place) = (tmp + epsilon).sqrt ().inverse ();
You can’t perform that action at this time.
0 commit comments