2 * Copyright (c) 2021 Samsung Electronics Co., Ltd. All Rights Reserved
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
19 #include "kernels/While.h"
21 namespace luci_interpreter
24 std::unique_ptr<Kernel> build_kernel_CircleWhile(const luci::CircleNode *circle_node,
25 KernelBuilderHelper &helper)
27 const auto *node = dynamic_cast<const luci::CircleWhile *>(circle_node);
29 throw std::runtime_error("wrong builder for operation");
31 auto output_nodes = collectOutputNodes<luci::CircleWhileOut>(node);
32 assert(node->arity() == node->input_count());
33 assert(output_nodes.size() == static_cast<size_t>(node->output_count()));
35 std::vector<const Tensor *> inputs(node->input_count());
36 for (uint32_t i = 0; i < node->input_count(); ++i)
38 inputs[i] = helper.getInputTensor(node->input(i));
40 std::vector<Tensor *> outputs = helper.getOutputTensors(output_nodes);
42 RuntimeGraph *cond_graph = helper.getRuntimeGraph(node->cond_graph());
43 RuntimeGraph *body_graph = helper.getRuntimeGraph(node->body_graph());
45 return std::make_unique<kernels::While>(std::move(inputs), std::move(outputs), cond_graph,
49 } // namespace luci_interpreter