|
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 |
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. | |