ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Rank.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 "Rank.h"
18
19
namespace
tflchef
20
{
21
22
void
TFliteOpRank::filler
(
const
tflite::Operator *op,
TFliteImport
*
import
,
23
tflchef::ModelRecipe *model_recipe)
const
24
{
25
// Nothing to do with filler
26
}
27
28
tflchef::Operation *
TFliteOpRank::build
(
RecipeChefContext
*ctx)
const
29
{
30
tflchef::Operation *operation = ctx->
chefop
;
31
const
tflite::Operator *op = ctx->
tflop
;
32
33
operation->set_type(
"Rank"
);
34
35
return
operation;
36
}
37
38
}
// namespace tflchef
tflchef::TFliteImport
Loads TF lite file and provides helpers to access attributes.
Definition
TFliteImport.h:41
tflchef::TFliteOpRank::build
tflchef::Operation * build(RecipeChefContext *ctx) const override
Definition
Rank.cpp:28
tflchef::TFliteOpRank::filler
void filler(const tflite::Operator *op, TFliteImport *import, tflchef::ModelRecipe *model_recipe) const override
Definition
Rank.cpp:22
tflchef
Definition
ModelChef.h:25
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
Rank.cpp
Generated by
1.9.8