Skip to content

Commit ce881a4

Browse files
authored
[luci/service] Enable CircleOutput shape inference (#14845)
This commit enables shape inference for CircleOutput node. ONE-DCO-1.0-Signed-off-by: Mateusz Bencer <m.bencer@partner.samsung.com>
1 parent bb33790 commit ce881a4

2 files changed

Lines changed: 34 additions & 1 deletion

File tree

compiler/luci/service/include/luci/Service/CircleShapeInference.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ class Algorithm final : public luci::CircleNodeVisitor<loco::TensorShape>
173173
// loco::TensorShape visit(const luci::CircleInput *node) final;
174174
// loco::TensorShape visit(const luci::CircleNonMaxSuppressionV4Out *node) final;
175175
// loco::TensorShape visit(const luci::CircleNonMaxSuppressionV5Out *node) final;
176-
// loco::TensorShape visit(const luci::CircleOutput *node) final;
176+
loco::TensorShape visit(const luci::CircleOutput *node) final;
177177
// loco::TensorShape visit(const luci::CircleOutputDummy *node) final;
178178
// loco::TensorShape visit(const luci::CircleOutputExclude *node) final;
179179
// loco::TensorShape visit(const luci::CircleSplitOut *node) final;
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
/*
2+
* Copyright (c) 2025 Samsung Electronics Co., Ltd. All Rights Reserved
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
17+
#include "luci/Service/CircleShapeInference.h"
18+
19+
#include "CircleShapeInferenceHelper.h"
20+
21+
namespace luci
22+
{
23+
namespace sinf
24+
{
25+
26+
loco::TensorShape Algorithm::visit(const luci::CircleOutput *node)
27+
{
28+
const auto from_shape = loco::must_cast<luci::CircleNode *>(node->from());
29+
return sinf::circle_shape(from_shape);
30+
}
31+
32+
} // namespace sinf
33+
} // namespace luci

0 commit comments

Comments
 (0)