io.h File Reference

#include <vector>
#include <string>
#include <utility>
#include <queue>
#include "dmlc/data.h"
#include "dmlc/registry.h"
#include "./base.h"
#include "./ndarray.h"

Include dependency graph for io.h:

Go to the source code of this file.

Classes
class	mxnet::IIterator< DType >
	iterator type More...
struct	mxnet::DataInst
	a single data instance More...
struct	mxnet::DataBatch
	DataBatch of NDArray, returned by Iterator. More...
struct	mxnet::DataIteratorReg
	Registry entry for DataIterator factory functions. More...

Namespaces
	mxnet
	namespace of mxnet

Macros
#define	MXNET_REGISTER_IO_ITER(name)   DMLC_REGISTRY_REGISTER(::mxnet::DataIteratorReg, DataIteratorReg, name)
	Macro to register Iterators. More...

Typedefs
typedef std::function< IIterator< DataBatch > *()>	mxnet::DataIteratorFactory
	typedef the factory function of data iterator More...

Macro Definition Documentation

#define MXNET_REGISTER_IO_ITER

(

name

)

DMLC_REGISTRY_REGISTER(::mxnet::DataIteratorReg, DataIteratorReg, name)

Macro to register Iterators.

// example of registering a mnist iterator
REGISTER_IO_ITE(MNISTIter)
.describe("Mnist data iterator")
.set_body([]() {
    return new PrefetcherIter(new MNISTIter());
  });