ONE - On-device Neural Engine
Loading...
Searching...
No Matches
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.