Skip to content
 
 

Repository files navigation

Torch2H: PyTorch Graph -> H Program

Requirements

  • node.js >= 12.x
  • python >= 3.8

How to build and use

# install dependencies
npm run install:all

# build
npm run build

# running example: Generates H program "out.h".
python torch2h.py examples/mnist/main.py

TODO

  • Make realistic weights by learning network. Current weights are just randomly generated floats.
  • Add more benchmarks and functions accordingly.

Etc.

This program has built on PyTea.

About

PyTEA: PyTorch Tensor Error Analyzer

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages