Crate hpt

Crate hpt 

Source
Expand description

This crate is dynamic graph based tensor library

Modules§

alloc
module for memory allocation
backend
module for backend
buitin_templates
module for buitin templates
common
module for common utils like shape and strides
error
module for error handling
iter
module for tensor iterator
ops
this module contains all the operators for the Tensor
re_exports
reexport serde
save_load
module for save and load
types
type related module
utils
module for utils, like set_num_threads, set_seed, etc.

Macros§

conv2d_micro_kernel
conv2d micro kernel
conv2d_micro_kernel_with_padding
conv2d micro kernel with padding
conv2d_mixed_precision_micro_kernel
conv2d mixed precision micro kernel
conv2d_mixed_precision_micro_kernel_with_padding
conv2d mixed precision micro kernel
define_matmul_micro_kernel
Define Matmul Micro Kernel function
define_mixed_precision_matmul_micro_kernel
Define Matmul Micro Kernel function
define_mixed_precision_post_op_matmul_micro_kernel
Define Matmul Micro Kernel function with post operation
define_post_op_matmul_micro_kernel
Define Matmul Micro Kernel function with post operation
slice
slice operation for tensor slicing uses the same syntax as numpy

Structs§

Tensor
Tensor is alias of N-dimensional array.

Traits§

Load
Save

Derive Macros§

Load
generate load trait
Save
generate save trait