ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Nodes.h File Reference

Go to the source code of this file.

Data Structures

class  loco::Push
 Make a value visible to user. More...
 
class  loco::Pull
 Create a value from user data. More...
 
class  loco::Forward
 Create a new value identical to its input. More...
 
class  loco::ReLU
 Create a new value that rectifies its input. More...
 
class  loco::ReLU6
 Create a new value that rectifies its input capping the units at 6. More...
 
class  loco::Tanh
 Create a new value that rectifies its input by tanh. More...
 
class  loco::ConstGen
 Create a value from constant byte array. More...
 
class  loco::MaxPool2D
 2D Max Pooling More...
 
class  loco::AvgPool2D
 2D Average Pooling More...
 
class  loco::FeatureEncode
 Create a feature map from a tensor. More...
 
class  loco::FeatureDecode
 Create a tensor from a feature map. More...
 
class  loco::FilterEncode
 Create a filter from a tensor. More...
 
class  loco::FilterDecode
 Create a tensor from a filter. More...
 
class  loco::DepthwiseFilterEncode
 Create a depthwise filter from a tensor. More...
 
class  loco::DepthwiseFilterDecode
 Create a tensor from a depthwise filter. More...
 
class  loco::Reshape< ReshapeType::Fixed >
 Reshape a tensor to another tensor whose shape is known at compile time. More...
 
class  loco::TensorConcat
 Concatenate two tensors. More...
 
class  loco::Conv2D
 2D Spatial Convolution More...
 
class  loco::DepthwiseConv2D
 Depthwise 2D Convolution. More...
 
class  loco::TensorReduce
 Computes ReduceFunc operations for Tensor domain. More...
 
class  loco::TransposedConv2D
 2D Transposed Convolution More...
 
class  loco::Softmax< Domain::Tensor >
 Computes softmax activations for Tensor domain. More...
 
class  loco::BiasDecode
 Create a "Tensor" from a "Bias". More...
 
class  loco::BiasEncode
 Create a "Bias" from a "Tensor". More...
 
class  loco::BiasAdd< Domain::Tensor >
 Add Tensor and Bias. More...
 
class  loco::BiasAdd< Domain::Feature >
 Add Feature and Bias along "depth" axis. More...
 
class  loco::TensorConstantPad
 Pads a tensor with constant value. More...
 
class  loco::EltwiseAdd
 Elementwise Add lhs and rhs. More...
 
class  loco::EltwiseMax
 Elementwise Maximum of lhs and rhs. More...
 
class  loco::EltwiseMul
 Elementwise Mul lhs and rhs. More...
 
class  loco::EltwiseSub
 Elementwise Sub lhs and rhs. More...
 
class  loco::EltwiseDiv
 Elementwise Div lhs and rhs. More...
 
class  loco::EltwiseSqrt
 Elementwise Sqrt of input. More...
 
class  loco::TensorBroadcast
 Duplicate elements along specified axes. More...
 
class  loco::TensorBroadcast::Mapping
 
class  loco::MatrixEncode
 Create Matrix from Tensor. More...
 
class  loco::MatrixDecode
 Create Tensor from Matrix. More...
 
class  loco::MatMul
 Matrix Multiplication lhs and rhs. More...
 
class  loco::TensorTranspose
 Permute an input. More...
 
class  loco::TensorTranspose::Perm
 

Namespaces

namespace  loco
 

Typedefs

using loco::FixedReshape = Reshape< ReshapeType::Fixed >
 
using loco::TensorSoftmax = Softmax< Domain::Tensor >
 
using loco::TensorBiasAdd = BiasAdd< Domain::Tensor >
 
using loco::FeatureBiasAdd = BiasAdd< Domain::Feature >
 

Enumerations

enum class  loco::ReshapeType { loco::Fixed }
 
enum class  loco::ReduceFunc { loco::Mean }
 Reduce type functions. More...
 

Functions

void loco::link (GraphOutput *, Push *push)
 
Pushloco::push_node (Graph *g, const GraphOutputIndex &index)
 Find a Push node with a given output index.
 
void loco::link (GraphInput *, Pull *pull)
 
Pullloco::pull_node (Graph *g, const GraphInputIndex &index)
 Find a Pull node with a given input index.