ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleExporterUtils.h File Reference
#include "ExporterUtils.h"
#include "circle_schema_generated.h"
#include "Dialect/IR/TFLNodes.h"
#include <loco.h>
#include <unordered_map>

Go to the source code of this file.

Data Structures

struct  exo::circle_detail::OpCode
 
struct  std::hash< exo::circle_detail::OpCode >
 
struct  exo::circle_detail::SubGraphContext
 Record the information of T/F Lite SubGraph and its mapping to loco. More...
 
struct  exo::circle_detail::SerializedModelData
 

Namespaces

namespace  exo
 
namespace  exo::circle_detail
 
namespace  std
 

Typedefs

using exo::circle_detail::TFLTensorIndex = int32_t
 

Functions

circle::ActivationFunctionType exo::to_circle_actfunc (locoex::FusedActFunc func)
 
circle::Padding exo::circle_detail::getOpPadding (const loco::Padding2D *pad, const loco::Stride< 2 > *stride, const ShapeDescription &ifm, const ShapeDescription &ofm)
 
circle::Padding exo::circle_detail::getOpPadding (const locoex::Padding pad)
 
void exo::circle_detail::registerGraphIOName (loco::Graph *graph, SerializedModelData &gd)
 Register graph input and output names to SerializedModelData.
 
void exo::circle_detail::set_tensor_index (loco::Node *node, const TFLTensorIndex &tensor_id)
 
TFLTensorIndex exo::circle_detail::get_tensor_index (loco::Node *node)