Caffe2 - C++ API
A deep learning, cross platform ML framework
datasets.h
1 #pragma once
2 
3 #include <torch/data/datasets/base.h>
4 #include <torch/data/datasets/chunk.h>
5 #include <torch/data/datasets/map.h>
6 #include <torch/data/datasets/mnist.h>
7 #include <torch/data/datasets/shared.h>
8 #include <torch/data/datasets/stateful.h>
9 #include <torch/data/datasets/tensor.h>