ONE - On-device Neural Engine
|
#include "loco/IR/Node.h"
#include "loco/IR/Use.h"
#include "loco/IR/Domain.h"
#include "loco/IR/DataType.h"
#include "loco/IR/DataTypeTraits.h"
#include "loco/IR/Dimension.h"
#include "loco/IR/Window.h"
#include "loco/IR/Stride.h"
#include "loco/IR/Padding2D.h"
#include "loco/IR/PaddingND.h"
#include "loco/IR/TensorAxis.h"
#include "loco/IR/TensorAxisSet.h"
#include "loco/IR/FeatureCodec.h"
#include "loco/IR/FilterCodec.h"
#include "loco/IR/DepthwiseFilterCodec.h"
#include "loco/IR/MatrixCodec.h"
#include "loco/IR/NodeMixins.h"
#include "loco/IR/CanonicalNodeDecl.h"
#include "loco/IR/GraphInputIndex.h"
#include "loco/IR/GraphOutputIndex.h"
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) |
Push * | loco::push_node (Graph *g, const GraphOutputIndex &index) |
Find a Push node with a given output index. | |
void | loco::link (GraphInput *, Pull *pull) |
Pull * | loco::pull_node (Graph *g, const GraphInputIndex &index) |
Find a Pull node with a given input index. | |