Module Expr Str Import alias Import alias identifier ImportFrom alias Assign NameStore Call AttributeLoad NameLoad attr NameLoad ClassDef bases NameLoad body Expr Str FunctionDef arguments args NameParam NameParam defaults NameLoad body Expr Call AttributeLoad Call NameLoad NameLoad NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam defaults body Raise Call NameLoad decorator_list FunctionDef arguments args NameParam defaults body Raise Call NameLoad decorator_list FunctionDef arguments args NameParam defaults body Raise Call NameLoad decorator_list decorator_list ClassDef bases NameLoad body Expr Str FunctionDef arguments args NameParam NameParam NameParam NameParam NameParam NameParam NameParam defaults NameLoad NameLoad NameLoad NameLoad NameLoad body Expr Str Expr Call AttributeLoad Call NameLoad NameLoad NameLoad attr keyword NameLoad Assign NameStore IfExp Call NameLoad NameLoad NameLoad NameLoad ListLoad NameLoad Assign AttributeStore NameLoad attr Call NameLoad SubscriptLoad NameLoad Index Num Assert CompareGtE AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr Assign AttributeStore NameLoad attr Num Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr NameLoad If BoolOpAnd NameLoad CompareIs NameLoad NameLoad CompareIsNot NameLoad NameLoad body Raise Call NameLoad Str Assign AttributeStore NameLoad attr ListComp IfExp CompareIs NameLoad NameLoad SubscriptLoad AttributeLoad NameLoad attr Index Num NameLoad comprehension NameStore NameLoad If CompareEq Call NameLoad AttributeLoad NameLoad attr Num body Assign AttributeStore NameLoad attr SubscriptLoad AttributeLoad NameLoad attr Index Num Assign AttributeStore NameLoad attr NameLoad FunctionDef arguments args NameParam defaults body Return TupleLoad Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Call AttributeLoad AttributeLoad NameLoad attr attr SubscriptLoad AttributeLoad NameLoad attr Index Num Lambda arguments args NameParam NameParam defaults Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad NameLoad decorator_list FunctionDef arguments args NameParam defaults body Return TupleLoad Call AttributeLoad AttributeLoad NameLoad attr attr Call AttributeLoad NameLoad attr TupleLoad Num Num keyword AttributeLoad NameLoad attr Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Lambda arguments args NameParam NameParam defaults Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad keyword Num keyword NameLoad decorator_list Assign TupleStore AttributeStore NameLoad attr AttributeStore NameLoad attr AttributeStore NameLoad attr Call NameLoad ListComp Call NameLoad NameLoad comprehension NameStore NameLoad Assign TupleStore AttributeStore NameLoad attr AttributeStore NameLoad attr TupleLoad Call NameLoad AttributeLoad NameLoad attr Call NameLoad AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call NameLoad AttributeLoad NameLoad attr If CompareIsNot NameLoad NameLoad body Assign TupleStore AttributeStore NameLoad attr AttributeStore NameLoad attr NameStore IfExp NameLoad Call NameLoad NameLoad Call NameLoad NameLoad Expr Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr Expr Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr Expr Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad decorator_list FunctionDef arguments args NameParam defaults body Return UnaryOpUSub BinOpFloorDiv BinOpSub AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr decorator_list NameLoad FunctionDef arguments args NameParam defaults body Expr Str Assign AttributeStore NameLoad attr Num decorator_list FunctionDef arguments args NameParam defaults body Expr Str For NameStore Call NameLoad AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr body Assign NameStore Call NameLoad AttributeLoad AttributeLoad NameLoad attr attr BinOpSub AttributeLoad NameLoad attr NameLoad Assign TupleStore NameStore NameStore TupleLoad Call NameLoad Num NameLoad Call NameLoad NameLoad BinOpAdd NameLoad NameLoad Assign TupleStore NameStore NameStore TupleLoad NameLoad NameLoad If CompareGt AttributeLoad AttributeLoad NameLoad attr attr NameLoad body Assign TupleStore NameStore NameStore TupleLoad Call NameLoad NameLoad NameLoad Call NameLoad Num BinOpSub AttributeLoad AttributeLoad NameLoad attr attr NameLoad Assign AttributeStore NameLoad attr BinOpSub AttributeLoad AttributeLoad NameLoad attr attr NameLoad For TupleStore NameStore NameStore NameStore Call NameLoad AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr body Expr Call NameLoad SubscriptLoad NameLoad Index NameLoad SubscriptLoad NameLoad ExtSlice Slice Index NameLoad If NameLoad body Expr Call NameLoad SubscriptLoad NameLoad Index NameLoad SubscriptLoad NameLoad ExtSlice Slice Index NameLoad Assign NameStore IfExp CompareEq Call NameLoad AttributeLoad NameLoad attr Num SubscriptLoad AttributeLoad NameLoad attr Index Num AttributeLoad NameLoad attr Assign NameStore IfExp AttributeLoad NameLoad attr AttributeLoad NameLoad attr NameLoad Expr Yield TupleLoad NameLoad NameLoad decorator_list decorator_list ClassDef bases NameLoad body Expr Str FunctionDef arguments args NameParam defaults vararg kwarg body Expr Call AttributeLoad NameLoad attr Str Expr Call AttributeLoad Call NameLoad NameLoad NameLoad attr NameLoad NameLoad decorator_list decorator_list If CompareEq NameLoad Str body ImportFrom alias Assign TupleStore TupleStore NameStore NameStore TupleStore NameStore NameStore Call NameLoad ImportFrom alias Assign NameStore Call NameLoad Num keyword Num Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr Num Assign NameStore Call NameLoad ListLoad SubscriptLoad NameLoad Slice Num SubscriptLoad NameLoad Slice Num SubscriptLoad NameLoad Slice Num keyword Num For NameStore Call NameLoad Num body For TupleStore NameStore TupleStore NameStore NameStore Call NameLoad NameLoad body Print NameLoad AttributeLoad NameLoad attr Pass
N/A N/A 
Defines basic input datatset types.
 N/A logging N/A numpy np neon NervanaObject N/A logger N/A N/A logging getLogger __name__ NervanaDataIterator N/A NervanaObject N/A N/A 
    Abstract class for data iterators.

    For serialization, any data iterator should inherit from this class
     __init__ N/A N/A self name N/A None N/A N/A N/A N/A N/A super NervanaDataIterator self __init__ name N/A nbatches N/A N/A self N/A N/A N/A N/A NotImplemented N/A reset N/A N/A self N/A N/A N/A N/A NotImplemented N/A __iter__ N/A N/A self N/A N/A N/A N/A NotImplemented N/A N/A ArrayIterator N/A NervanaDataIterator N/A N/A 
    This generic class defines an interface to iterate over minibatches of
    data that has been preloaded into memory in the form of numpy arrays.
    This may be used when the entire dataset is small enough to fit within memory.
     __init__ N/A N/A self X y nclass lshape make_onehot name N/A None None None True None N/A N/A 
        Implements loading of given data into backend tensor objects. If the
        backend is specific to an accelarator device, the data is copied over
        to that device.

        Args:
            X (ndarray, shape: [# examples, feature size]): Input features within the
                dataset.
            y (ndarray, shape:[# examples, 1], optional): Labels corresponding to the
                input features.
                If absent, the input features themselves will be returned as
                target values (AutoEncoder)
            nclass (int, optional): The number of possible types of labels.
                (not necessary if not providing labels)
            lshape (tuple, optional): Local shape for the input features
                (e.g. height, width, channel for images)
            make_onehot (bool, optional): True if y is a label that has to be converted to one hot
                            False if y doesn't need to be converted to one hot
                            (e.g. in a CAE)

         N/A N/A N/A N/A super ArrayIterator self __init__ name name N/A X N/A N/A isinstance X list X N/A X N/A N/A self ndata N/A len N/A X N/A 0 N/A N/A N/A self ndata N/A N/A self be bsz N/A N/A self start 0 N/A N/A self nclass nclass N/A N/A self ybuf None N/A N/A make_onehot N/A nclass None N/A y None N/A N/A N/A AttributeError Must provide number of classes when creating onehot labels N/A N/A self shape N/A N/A N/A lshape None N/A N/A x shape N/A 1 lshape N/A x X N/A N/A N/A len N/A self shape 1 N/A N/A N/A self shape N/A N/A self shape N/A 0 N/A N/A self lshape lshape transpose_gen N/A N/A z N/A N/A N/A N/A N/A N/A N/A self be array z N/A N/A N/A self be iobuf N/A N/A z shape N/A 1 N/A N/A N/A _in _out N/A N/A N/A N/A self be copy_transpose _in _out N/A onehot_gen N/A N/A z N/A N/A N/A N/A N/A N/A N/A self be array N/A N/A z reshape N/A -1 1 dtype N/A np int32 N/A N/A N/A self be iobuf nclass N/A N/A N/A _in _out N/A N/A N/A N/A self be onehot _in axis 0 out _out N/A N/A N/A N/A self Xdev N/A self Xbuf N/A self unpack_func N/A zip N/A N/A transpose_gen x N/A x X N/A N/A N/A self dbuf N/A self hbuf N/A N/A list N/A self Xdev N/A list N/A self Xbuf N/A N/A self unpack_func N/A list N/A self unpack_func N/A N/A y None N/A N/A N/A N/A self ydev N/A self ybuf yfunc N/A make_onehot N/A onehot_gen y N/A transpose_gen y N/A N/A N/A N/A self dbuf append N/A self ydev N/A N/A N/A N/A self hbuf append N/A self ybuf N/A N/A N/A N/A self unpack_func append yfunc N/A nbatches N/A N/A self N/A N/A N/A N/A N/A N/A N/A self start N/A self ndata N/A N/A self be bsz N/A property reset N/A N/A self N/A N/A N/A 
        For resetting the starting index of this dataset back to zero.
        Relevant for when one wants to call repeated evaluations on the dataset
        but don't want to wrap around for the last uneven minibatch
        Not necessary when ndata is divisible by batch size
         N/A N/A self start 0 N/A __iter__ N/A N/A self N/A N/A N/A 
        Defines a generator that can be used to iterate over this dataset.

        Yields:
            tuple: The next minibatch which includes both features and labels.
         N/A i1 N/A range N/A self start N/A self ndata N/A N/A self be bsz N/A N/A bsz N/A min N/A N/A self be bsz N/A N/A self ndata i1 N/A N/A islice1 oslice1 N/A N/A slice 0 bsz N/A slice i1 N/A i1 bsz N/A N/A islice2 oslice2 N/A None None N/A N/A N/A N/A self be bsz bsz N/A N/A N/A islice2 oslice2 N/A N/A slice bsz None N/A slice 0 N/A N/A N/A self be bsz bsz N/A N/A self start N/A N/A N/A self be bsz bsz N/A N/A buf dev unpack_func N/A zip N/A self hbuf N/A self dbuf N/A self unpack_func N/A N/A N/A unpack_func N/A dev N/A oslice1 N/A buf N/A N/A N/A islice1 N/A oslice2 N/A N/A N/A unpack_func N/A dev N/A oslice2 N/A buf N/A N/A N/A islice2 N/A inputs N/A N/A N/A len N/A self Xbuf 1 N/A N/A self Xbuf N/A 0 N/A self Xbuf N/A targets N/A N/A self ybuf N/A self ybuf inputs N/A N/A N/A inputs targets N/A N/A DataIterator N/A ArrayIterator N/A N/A 
    This class has been renamed to ArrayIterator and deprecated.
    This is just a place holder until the class is removed.  Please
    use the ArrayIterator class.
     __init__ N/A N/A self N/A args kwargs N/A N/A N/A N/A logger error DataIterator class has been deprecated and renamed"ArrayIterator" please use that name. N/A N/A N/A N/A super DataIterator self __init__ args kwargs N/A N/A N/A N/A __name__ __main__ N/A neon.data load_mnist N/A N/A N/A X_train y_train N/A X_test y_test N/A load_mnist neon.backends.nervanagpu NervanaGPU N/A ng N/A NervanaGPU 0 device_id 1 N/A N/A NervanaObject be ng N/A N/A ng bsz 128 N/A train_set N/A ArrayIterator N/A N/A X_test N/A 1000 N/A X_test N/A 1000 N/A y_test N/A 1000 nclass 10 N/A i N/A range 3 N/A N/A N/A bidx N/A X_batch y_batch N/A enumerate train_set N/A N/A bidx N/A train_set start N/A