ONE - On-device Neural Engine
Loading...
Searching...
No Matches
FloorDiv.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2020 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
#ifndef LUCI_INTERPRETER_KERNELS_FLOOR_DIV_H
18
#define LUCI_INTERPRETER_KERNELS_FLOOR_DIV_H
19
20
#include "
core/Kernel.h
"
21
22
namespace
luci_interpreter
23
{
24
namespace
kernels
25
{
26
27
class
FloorDiv
:
public
Kernel
28
{
29
public
:
30
FloorDiv
(
const
Tensor
*
x
,
const
Tensor
*
y
,
Tensor
*
output
);
31
32
const
Tensor
*
x
()
const
{
return
_inputs
[0]; }
33
const
Tensor
*
y
()
const
{
return
_inputs
[1]; }
34
Tensor
*
output
()
const
{
return
_outputs
[0]; }
35
36
void
configure
()
override
;
37
void
execute
()
const override
;
38
39
private
:
40
void
evalFloat()
const
;
41
};
42
43
}
// namespace kernels
44
}
// namespace luci_interpreter
45
46
#endif
// LUCI_INTERPRETER_KERNELS_FLOOR_DIV_H
Kernel.h
luci_interpreter::Kernel
Definition
Kernel.h:29
luci_interpreter::Kernel::_inputs
const std::vector< const Tensor * > _inputs
Definition
Kernel.h:52
luci_interpreter::Kernel::_outputs
const std::vector< Tensor * > _outputs
Definition
Kernel.h:53
luci_interpreter::Tensor
Definition
Tensor.h:101
luci_interpreter::kernels::FloorDiv
Definition
FloorDiv.h:28
luci_interpreter::kernels::FloorDiv::execute
void execute() const override
Definition
FloorDiv.cpp:43
luci_interpreter::kernels::FloorDiv::y
const Tensor * y() const
Definition
FloorDiv.h:33
luci_interpreter::kernels::FloorDiv::configure
void configure() override
Definition
FloorDiv.cpp:35
luci_interpreter::kernels::FloorDiv::output
Tensor * output() const
Definition
FloorDiv.h:34
luci_interpreter::kernels::FloorDiv::x
const Tensor * x() const
Definition
FloorDiv.h:32
luci_interpreter
Definition
BuddyMemoryManager.h:22
compiler
luci-interpreter
src
kernels
FloorDiv.h
Generated by
1.9.8