ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Less.cpp
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
#include "Less.h"
18
19
#include "Convert.h"
20
#include "
FillerHelper.h
"
21
22
namespace
tflchef
23
{
24
25
void
TFliteOpLess::filler
(
const
tflite::Operator *op,
TFliteImport
*
import
,
26
tflchef::ModelRecipe *model_recipe)
const
27
{
28
const
auto
&inputs = *op->inputs();
29
30
for
(
int
input : inputs)
31
{
32
fill_tensor_to_import
(input,
import
);
33
}
34
}
35
36
tflchef::Operation *
TFliteOpLess::build
(
RecipeChefContext
*ctx)
const
37
{
38
tflchef::Operation *operation = ctx->
chefop
;
39
const
tflite::Operator *op = ctx->
tflop
;
40
41
operation->set_type(
"Less"
);
42
43
return
operation;
44
}
45
46
}
// namespace tflchef
FillerHelper.h
tflchef::TFliteImport
Loads TF lite file and provides helpers to access attributes.
Definition
TFliteImport.h:41
tflchef::TFliteOpLess::build
tflchef::Operation * build(RecipeChefContext *ctx) const override
Definition
Less.cpp:36
tflchef::TFliteOpLess::filler
void filler(const tflite::Operator *op, TFliteImport *import, tflchef::ModelRecipe *model_recipe) const override
Definition
Less.cpp:25
tflchef
Definition
ModelChef.h:25
tflchef::fill_tensor_to_import
void fill_tensor_to_import(int32_t idx, TFliteImport *import)
Definition
FillerHelper.cpp:24
tflchef::RecipeChefContext
Definition
TFliteOpChef.h:30
tflchef::RecipeChefContext::chefop
tflchef::Operation * chefop
Definition
TFliteOpChef.h:32
tflchef::RecipeChefContext::tflop
const tflite::Operator * tflop
Definition
TFliteOpChef.h:31
compiler
tflchef
tflite
src
Op
Less.cpp
Generated by
1.9.8