|
docc 0.8.3 (0d44999b)
C++ API Reference
|
Classes | |
| class | ArangeNode |
| Tensor node that generates a sequence of numbers. More... | |
| class | ArangeNodeSerializer |
| class | BatchNormNode |
| In N, C, D0...Dn tensor, applies normalization to per channel in C across all the inner dimensions. More... | |
| class | BatchNormNodeSerializer |
| class | BroadcastNode |
| class | BroadcastNodeSerializer |
| class | ClampNode |
| Computes Y = min(max(X, min), max) elementwise, where either bound may be omitted. More... | |
| class | ClampNodeSerializer |
| class | ConcatNode |
| Tensor concatenation node that can concatenate a number of tensors into another tensor. More... | |
| class | ConcatNodeSerializer |
| class | ConditionalTensorCopyNode |
| Conditional tensor copy node that copies values from one tensor or another depending on a tensor mask. More... | |
| class | ConditionalTensorCopyNodeSerializer |
| class | ConstPaddingNode |
| Applies padding by a constant value. More... | |
| class | ConstPaddingNodeSerializer |
| class | ConvNode |
| Convolution operation following ONNX Conv operator specification. More... | |
| class | ConvNodeSerializer |
| Serializer for ConvNode. More... | |
| class | TensorCopyNode |
| Tensor copy node that copies values form one container to another according to their tensor access. More... | |
| class | TensorCopyNodeSerializer |
| class | ElementWiseDataflowTensorNode |
| Defined by each element of the output tensor (defined by shape) can be defined independently of any other output. More... | |
| class | BaseElementWiseDataflowTensorNodeSerializer |
| class | SimpleElementWiseDataflowTensorNodeSerializer |
| class | AbsNode |
| class | AddNode |
| class | CastNode |
| class | CMathTensorNode |
| class | CMathTensorNodeSerializer |
| class | DivNode |
| class | ErfNode |
| class | ExpNode |
| class | FillNode |
| class | GELUNode |
| Applies the Gaussion Error Linear Units function. More... | |
| class | GELUNodeSerializer |
| class | LogicalNotNode |
| class | MaximumNode |
| class | MinimumNode |
| class | MulNode |
| class | PowNode |
| class | ReLUNode |
| class | RsqrtNode |
| class | SigmoidNode |
| class | SqrtNode |
| class | SubNode |
| class | TanhNode |
| class | TaskletTensorNode |
| class | TaskletTensorNodeSerializer |
| class | EluNode |
| Computes Y = X < 0 ? More... | |
| class | EluNodeSerializer |
| class | EmbeddingNode |
| Embedding lookup node implementing aten.embedding(weight, indices). More... | |
| class | EmbeddingNodeSerializer |
| class | EmbeddingRenormNode |
| In-place embedding renormalization implementing aten.embedding_renorm_. More... | |
| class | EmbeddingRenormNodeSerializer |
| class | IndexNode |
| Tensor advanced-indexing node. More... | |
| class | IndexNodeSerializer |
| class | LayerNormNode |
| Applies layer normalization over the trailing noramlized_shape dimensions of the input. More... | |
| class | LayerNormNodeSerializer |
| class | LeakyReLUNode |
| Computes Y = X > 0 ? More... | |
| class | LeakyReLUNodeSerializer |
| class | MatMulNode |
| Tensor matrix multiplication following ONNX MatMul operator specification. More... | |
| class | MatMulNodeSerializer |
| Serializer for MatMulNode. More... | |
| class | MatMulNaiveExpander |
| class | MatMulBlasExpander |
| class | PoolingNode |
| N-dimensional pooling operation. More... | |
| class | PoolingNodeSerializer |
| Serializer for PoolingNode. More... | |
| class | ReduceNode |
| Base class for tensor reduction operations. More... | |
| class | ReduceNodeSerializer |
| class | MaxNode |
| class | MeanNode |
| class | MinNode |
| class | SoftmaxNode |
| class | StdNode |
| class | SumNode |
| class | SpatialTensorNode |
| Specialization of TensorNode for spatial operations such as Convolution and Pooling. More... | |
| class | SpatialTensorNodeBaseSerializer |
| struct | InputContainerInfo |
| struct | MapDimension |
| class | TensorLayout |
| The metadata associated with a addressing a tensor in elements Meant to be used with TensorNodes to describe their input's and outputs layouts and ease handling of such descriptions If the TensorType will keep existing it should also be switched to use this. More... | |
| struct | ContractionTypes |
| Element types of a contraction such as a matmul. More... | |
| class | TensorNode |
| Abstract base class for all tensor operations. More... | |
| class | UpsampleBilinear2DNode |
| 2D bilinear upsampling operation More... | |
| class | UpsampleBilinear2DNodeSerializer |
| Serializer for UpsampleBilinear2DNode. More... | |
Enumerations | |
| enum class | PoolingMode { Max , Sum , Avg } |
| Pooling operation type. More... | |
Variables | |
| constexpr QuantizationType | QUANTIZATION_MATCH_INPUTS = types::PrimitiveType::Void |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<AbsNode> sdfg::math::tensor::AbsNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<AddNode> sdfg::math::tensor::AddNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<CastNode> sdfg::math::tensor::CastNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<DivNode> sdfg::math::tensor::DivNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<ErfNode> sdfg::math::tensor::ErfNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<ExpNode> sdfg::math::tensor::ExpNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<FillNode> sdfg::math::tensor::FillNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<LogicalNotNode> sdfg::math::tensor::LogicalNotNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<MaximumNode> sdfg::math::tensor::MaximumNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<MinimumNode> sdfg::math::tensor::MinimumNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<MulNode> sdfg::math::tensor::MulNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<PowNode> sdfg::math::tensor::PowNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<ReLUNode> sdfg::math::tensor::ReLUNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<RsqrtNode> sdfg::math::tensor::RsqrtNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<SigmoidNode> sdfg::math::tensor::SigmoidNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<SqrtNode> sdfg::math::tensor::SqrtNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<SubNode> sdfg::math::tensor::SubNodeSerializer |
| typedef SimpleElementWiseDataflowTensorNodeSerializer<TanhNode> sdfg::math::tensor::TanhNodeSerializer |
|
strong |
| std::vector< MapDimension > sdfg::math::tensor::create_maps | ( | builder::StructuredSDFGBuilder & | builder, |
| const std::vector< symbolic::Expression > & | shape, | ||
| structured_control_flow::Sequence & | parent_seq ) |
| std::string sdfg::math::tensor::create_temp_var | ( | builder::StructuredSDFGBuilder & | builder, |
| const std::string & | prefix, | ||
| int | gen, | ||
| const types::IType & | type ) |
| types::PrimitiveType sdfg::math::tensor::deserialize_quantization | ( | const nlohmann::json & | j, |
| const std::string & | field_name, | ||
| types::PrimitiveType | default_value = QUANTIZATION_MATCH_INPUTS ) |
| InputContainerInfo sdfg::math::tensor::find_usable_input_access_node | ( | data_flow::DataFlowGraph & | dataflow, |
| data_flow::LibraryNode & | node, | ||
| const std::string & | input_conn ) |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
| std::ostream & sdfg::math::tensor::operator<< | ( | std::ostream & | stream, |
| const TensorLayout & | layout ) |
|
constexpr |