File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -667,7 +667,7 @@ InferCorrectLayoutOutput L2NormalizeInferCorrectLayout(const Attrs& attrs,
667667 old_in_shapes.push_back (old_in_t .as <TensorTypeNode>()->shape );
668668 }
669669 std::vector<size_t > axis_list;
670- for (auto i : param->axis ) {
670+ for (auto i : param->axis ) {
671671 int64_t axis = i->value ;
672672 if (axis < 0 ) {
673673 axis = axis + static_cast <size_t >(old_in_shapes[0 ].size ());
@@ -677,7 +677,7 @@ InferCorrectLayoutOutput L2NormalizeInferCorrectLayout(const Attrs& attrs,
677677
678678 Layout ret = Layout::Undef ();
679679 if (new_in_layouts.defined () && old_in_layouts.defined ()) {
680- for (uint i = 0 ; i < axis_list.size (); ++i) {
680+ for (uint i = 0 ; i < axis_list.size (); ++i) {
681681 const auto & axis_dim = old_in_layouts[0 ][axis_list[i]];
682682 auto axis_index = new_in_layouts[0 ].IndexOf (axis_dim);
683683 param->axis .Set (i, axis_index);
You can’t perform that action at this time.
0 commit comments