docc 0.8.3 (0d44999b)
C++ API Reference
Loading...
Searching...
No Matches
tensor Directory 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.