pytorch-inference
pytorch::inference_engine Member List

This is the complete list of members for pytorch::inference_engine, including all inherited members.

add_layer(Layer *l)pytorch::inference_engineinline
add_layer(std::vector< Layer *>l)pytorch::inference_engineinline
devicepytorch::inference_engineprivate
forward(const std::vector< tensor > &input)pytorch::inference_engineinline
get_layer_ptr(const int &depth, const int &width=0)pytorch::inference_engineinline
inference_engine(const int &device=0, af::Backend backend=AF_BACKEND_CUDA, bool quiet=true)pytorch::inference_engineinline
layerspytorch::inference_engineprivate
~inference_engine()pytorch::inference_engineinlinevirtual