#pragma once #include // There's no difference between aten, torch and caffe2 libs any more // TODO: clean up the naming for consistency #define TORCH_API CAFFE2_API