/*
- * Copyright (c) 2021 Samsung Electronics Co., Ltd. All Rights Reserved
+ * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
#include "Builders.h"
-#include "kernels/Floor.h"
+#include "kernels/FloorMod.h"
namespace luci_interpreter
{
-std::unique_ptr<Kernel> build_kernel_CircleFloor(std::vector<const Tensor *> &&inputs,
- std::vector<Tensor *> &&outputs,
- const uint32_t op_index, KernelBuilder &builder)
+std::unique_ptr<Kernel> build_kernel_CircleFloorMod(const luci::CircleNode *circle_node,
+ KernelBuilderHelper &helper)
{
- assert(inputs.size() == 1);
+ const auto *node = loco::must_cast<const luci::CircleFloorMod *>(circle_node);
+ assert(node->arity() == 2);
- const Tensor *input = inputs.at(0);
- Tensor *output = outputs.at(0);
+ const Tensor *x = helper.getInputTensor(node->x());
+ const Tensor *y = helper.getInputTensor(node->y());
+ Tensor *output = helper.getOutputTensor(node);
- return std::make_unique<kernels::Floor>(input, output);
+ return std::make_unique<kernels::FloorMod>(x, y, output);
}
} // namespace luci_interpreter