pytorch-inference
|
This is the complete list of members for pytorch::inference_engine, including all inherited members.
add_layer(Layer *l) | pytorch::inference_engine | inline |
add_layer(std::vector< Layer *>l) | pytorch::inference_engine | inline |
device | pytorch::inference_engine | private |
forward(const std::vector< tensor > &input) | pytorch::inference_engine | inline |
get_layer_ptr(const int &depth, const int &width=0) | pytorch::inference_engine | inline |
inference_engine(const int &device=0, af::Backend backend=AF_BACKEND_CUDA, bool quiet=true) | pytorch::inference_engine | inline |
layers | pytorch::inference_engine | private |
~inference_engine() | pytorch::inference_engine | inlinevirtual |