ONE - On-device Neural Engine
Loading...
Searching...
No Matches
tflchef::TFliteOpReverseV2 Class Reference

tflchef operator builder for ReverseV2 More...

#include <ReverseV2.h>

Collaboration diagram for tflchef::TFliteOpReverseV2:

Public Member Functions

void filler (const tflite::Operator *op, TFliteImport *import, tflchef::ModelRecipe *model_recipe) const override
 
tflchef::Operation * build (RecipeChefContext *ctx) const override
 
- Public Member Functions inherited from tflchef::TFliteOpChef
virtual ~TFliteOpChef ()
 

Detailed Description

tflchef operator builder for ReverseV2

Definition at line 28 of file ReverseV2.h.

Member Function Documentation

◆ build()

tflchef::Operation * tflchef::TFliteOpReverseV2::build ( RecipeChefContext ctx) const
overridevirtual

Implements tflchef::TFliteOpChef.

Definition at line 34 of file ReverseV2.cpp.

35{
36 tflchef::Operation *operation = ctx->chefop;
37 const tflite::Operator *op = ctx->tflop;
38
39 operation->set_type("ReverseV2");
40
41 return operation;
42}

References tflchef::RecipeChefContext::chefop, and tflchef::RecipeChefContext::tflop.

◆ filler()

void tflchef::TFliteOpReverseV2::filler ( const tflite::Operator *  op,
TFliteImport import,
tflchef::ModelRecipe *  model_recipe 
) const
overridevirtual

Implements tflchef::TFliteOpChef.

Definition at line 25 of file ReverseV2.cpp.

27{
28 const std::vector<int32_t> &inputs = as_index_vector(op->inputs());
29 assert(inputs.size() == 2);
30
31 fill_tensor_to_import(inputs[1], import);
32}
std::vector< T > as_index_vector(const flatbuffers::Vector< T > *flat_array)
Definition Convert.h:47
void fill_tensor_to_import(int32_t idx, TFliteImport *import)

References tflchef::as_index_vector(), and tflchef::fill_tensor_to_import().


The documentation for this class was generated from the following files: