pytorch-inference
include Directory Reference

Directories

directory  modules
 

Files

file  inference_engine.hpp [code]
 The engine that drives inferences. This will hold everything related to the OpenCL backend.
 
file  layers.hpp [code]
 Equivalent to Conv2d in pytorch.
 
file  py_object.hpp [code]
 
file  include/utils.hpp [code]