CaffeOnACL/include/caffe/caffe.hpp at master · 2php/CaffeOnACL · GitHub

Repository navigation

27 lines (23 loc) · 698 Bytes

// caffe.hpp is the header file that you need to include in your code. It wraps

// all the internal caffe header files into one for simpler inclusion.

#ifndef CAFFE_CAFFE_HPP_

#define CAFFE_CAFFE_HPP_

#ifdef USE_ACL

#ifndef CPU_ONLY

#define CPU_ONLY

#endif

#include "caffe/blob.hpp"

#include "caffe/common.hpp"

#include "caffe/filler.hpp"

#include "caffe/layer.hpp"

#include "caffe/layer_factory.hpp"

#include "caffe/net.hpp"

#include "caffe/parallel.hpp"

#include "caffe/proto/caffe.pb.h"

#include "caffe/solver.hpp"

#include "caffe/solver_factory.hpp"

#include "caffe/util/benchmark.hpp"

#include "caffe/util/io.hpp"

#include "caffe/util/upgrade_proto.hpp"

#endif // CAFFE_CAFFE_HPP_