Unverified Commit 7d831a2f authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Restructuring C++ project: (#3146)

Summary:
* Reduce unnecessary header inclusions in models and io.

* Move autocast to separate folder and hide autograd implementation in an anonymous namespace.

* Moving files in subfolders.

Reviewed By: fmassa

Differential Revision: D25461523

fbshipit-source-id: 756eeb6848aacaa474de4825ed4c1045d17e2cea
parent 4d3a3093
#include "read_image_cpu.h" #include "read_image_cpu.h"
#include "readjpeg_cpu.h" #include "readjpeg_cpu.h"
#include "readpng_cpu.h" #include "readpng_cpu.h"
......
#pragma once #pragma once
#include <torch/torch.h> #include <torch/types.h>
#include "image_read_mode.h" #include "../image_read_mode.h"
C10_EXPORT torch::Tensor decode_image( C10_EXPORT torch::Tensor decode_image(
const torch::Tensor& data, const torch::Tensor& data,
......
#pragma once #pragma once
#include <errno.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <torch/torch.h> #include <torch/types.h>
C10_EXPORT torch::Tensor read_file(const std::string& filename); C10_EXPORT torch::Tensor read_file(const std::string& filename);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment