inference_engine.hpp File Reference

The engine that drives inferences. This will hold everything related to the OpenCL backend. More...

#include <algorithm>
#include <functional>
#include <arrayfire.h>
#include "utils.hpp"
#include "layers.hpp"

Go to the source code of this file.

Classes
class	pytorch::inference_engine

Namespaces
	pytorch

Detailed Description

The engine that drives inferences. This will hold everything related to the OpenCL backend.