Skip to content

旨在学习CPU上实现深度学习推理框架,以及常用加速方法,参考KuiperInfer

Notifications You must be signed in to change notification settings

LudovicoYIN/InferNeto

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

InferNeto (自制深度学习推理框架)

  1. 实现基本Tensor类-cpu版本 Finnished
  2. 实现读取PNNX格式数据 Finnished
  3. 实现计算图类 Finnished
  4. 实现基本算子 Finnished
    1. 激活函数类:
      1. relu
      2. sigmoid
    2. 普通算子:
      1. convolution (im2col)
      2. linear
      3. maxpooling
      4. adaptive_avgpooling
      5. flatten
      6. softmax
    3. PNNX算子:
      1. expression
  5. 完成ResNet推理 Finnished
  6. 完成Yolo推理 TODO
  7. 性能调优 TODO
    1. Tilling技术、Simd指令集

About

旨在学习CPU上实现深度学习推理框架,以及常用加速方法,参考KuiperInfer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published