|
docc 0.8.3 (0d44999b)
C++ API Reference
|
Directories | |
| elementwise_ops | |
| reduce_ops | |
Files | |
| arange_node.h | |
| batchnorm_node.h | |
| broadcast_node.h | |
| clamp_node.h | |
| Clamp tensor node with PyTorch semantics. | |
| concat_node.h | |
| conditional_copy_node.h | |
| Conditional tensor copy node that copies values from one tensor or another depending on a tensor mask. | |
| const_padding_node.h | |
| conv_node.h | |
| Convolution operation node compatible with ONNX Conv operator. | |
| copy_node.h | |
| Tensor copy node that copies values from one buffer to another according to their tensor access. | |
| elementwise_node.h | |
| Tensor elementwise operation nodes. | |
| elu_node.h | |
| ELU tensor node with PyTorch semantics. | |
| embedding_node.h | |
| embedding_renorm_node.h | |
| index_node.h | |
| layernorm_node.h | |
| leaky_relu_node.h | |
| Leaky ReLU tensor node with PyTorch semantics. | |
| matmul_node.h | |
| Tensor matrix multiplication node compatible with ONNX MatMul operator. | |
| pooling_node.h | |
| reduce_node.h | |
| Tensor reduction operation nodes. | |
| spatial_tensor_node.h | |
| tensor_expansion_utils.h | |
| tensor_layout.h | |
| tensor_node.h | |
| Abstract base class for tensor library nodes. | |
| upsample_node.h | |
| Bilinear 2D upsampling node compatible with aten::upsample_bilinear2d.vec. | |