Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit b2296ae

Browse files
author
Hao Jin
committed
fix buggy InferType in L2Normalization
1 parent 5fc31b9 commit b2296ae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/operator/l2_normalization-inl.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -251,8 +251,8 @@ class L2NormalizationProp : public OperatorProperty {
251251
std::vector<int> *out_type,
252252
std::vector<int> *aux_type) const override {
253253
int dtype = (*in_type)[0];
254-
type_assign(&(*out_type)[0], dtype);
255-
type_assign(&(*out_type)[1], dtype);
254+
type_assign(&dtype, (*out_type)[0]);
255+
type_assign(&dtype, (*out_type)[1]);
256256

257257
TYPE_ASSIGN_CHECK(*in_type, 0, dtype);
258258
TYPE_ASSIGN_CHECK(*out_type, 0, dtype);

0 commit comments

Comments
 (0)