mxnet
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Classes | Namespaces | Macros | Typedefs
io.h File Reference

mxnet io data structure and data iterator More...

#include <dmlc/data.h>
#include <dmlc/registry.h>
#include <vector>
#include <string>
#include <utility>
#include <queue>
#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...
 

Detailed Description

mxnet io data structure and data iterator

Copyright (c) 2015 by Contributors

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());
});