mxnet
Classes | Namespaces | Macros | Typedefs
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.

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