Skip to main content

Module dispatch

Module dispatch 

Source
Expand description

Backend implementations of tensor operations.

Modules§

numeric
Numeric utility functions for jit backends

Functions§

empty_qtensor
Create an empty quantized tensor.
empty_qtensor_optimized
Create an empty quantized tensor.
into_data
into_data_sync
Read data from a RudaTensor synchronously
permute
Permute a tensor’s dimensions
permute_nchw_to_nhwc
Permute a tensor’s dimensions from NCHW to NHWC, or the N-dimensional equivalent
permute_nchw_to_nhwc_shape
Permute a shape’s dimensions from NCHW to NHWC, or the N-dimensional equivalent
permute_nhwc_to_nchw
Permute a tensor’s dimensions from NHWC to NCHW, or the N-dimensional equivalent
permute_nhwc_to_nchw_shape
Permute a shape’s dimensions from NHWC to NCHW, or the N-dimensional equivalent
q_reshape
Reshape a quantized tensor without recalibrating its values.
reshape
Reshape a jit tensor to a new shape
unfold
Unfold windows along a dimension.