Module Import alias identifier Import alias identifier ImportFrom alias alias alias alias ClassDef bases NameLoad body FunctionDef arguments args NameParam defaults body Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign AttributeStore NameLoad attr TupleLoad SubscriptLoad NameLoad Index Num Call AttributeLoad NameLoad attr SubscriptLoad NameLoad Slice Num Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign AttributeStore AttributeLoad NameLoad attr attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr decorator_list decorator_list ClassDef bases NameLoad body FunctionDef arguments args NameParam NameParam defaults body If UnaryOpNot Call NameLoad NameLoad NameLoad body If Call NameLoad NameLoad Str body Assign NameStore Call NameLoad NameLoad orelse Assign NameStore TupleLoad NameLoad If UnaryOpNot CompareLtELtE Num Call AttributeLoad Call NameLoad NameLoad attr Num Num body Raise Call NameLoad BinOpMod Str NameLoad Assign AttributeStore NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam defaults body Assign NameStore AttributeLoad AttributeLoad AttributeLoad NameLoad attr attr attr Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call NameLoad GeneratorExp IfExp CompareNotEq NameLoad Num NameLoad BinOpFloorDiv UnaryOpUSub NameLoad NameLoad comprehension NameStore AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign AttributeStore AttributeLoad NameLoad attr attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr decorator_list decorator_list ClassDef bases NameLoad NameLoad body FunctionDef arguments args NameParam NameParam defaults body Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr ListComp Call Call NameLoad NameLoad comprehension NameStore Call NameLoad BinOpAdd Call NameLoad AttributeLoad NameLoad attr Num decorator_list FunctionDef arguments args NameParam NameParam defaults body Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr ListLoad NameLoad Assign AttributeStore NameLoad attr AttributeLoad NameLoad attr For NameStore AttributeLoad NameLoad attr body Assign AttributeStore NameLoad attr AttributeLoad NameLoad attr Return AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign NameStore BinOpAdd BinOpAdd ListLoad Num AttributeLoad NameLoad attr ListLoad SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Index Num Assign AttributeStore NameLoad attr ListLoad For NameStore Call NameLoad BinOpSub Call NameLoad NameLoad Num body Expr Call AttributeLoad AttributeLoad NameLoad attr attr TupleLoad SubscriptLoad NameLoad Index NameLoad SubscriptLoad NameLoad Index BinOpAdd NameLoad Num For TupleStore NameStore TupleStore NameStore NameStore Call NameLoad AttributeLoad NameLoad attr body Assign NameStore BinOpAdd TupleLoad BinOpSub NameLoad NameLoad SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Slice Num Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr NameLoad Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Slice NameLoad NameLoad If AttributeLoad NameLoad attr body Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr Call AttributeLoad NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam defaults body For TupleStore NameStore TupleStore NameStore NameStore Call NameLoad AttributeLoad NameLoad attr body Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Slice NameLoad NameLoad decorator_list FunctionDef arguments args NameParam defaults body For TupleStore NameStore TupleStore NameStore NameStore Call NameLoad AttributeLoad NameLoad attr body Expr Call AttributeLoad NameLoad attr SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Slice NameLoad NameLoad AttributeLoad SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr decorator_list decorator_list ClassDef bases NameLoad body FunctionDef arguments args NameParam NameParam defaults NameLoad body Assign AttributeStore NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam defaults body If CompareEq Call NameLoad AttributeLoad AttributeLoad NameLoad attr attr Num body Assign AttributeStore NameLoad attr BinOpAdd TupleLoad Num AttributeLoad AttributeLoad NameLoad attr attr orelse If CompareEq Call NameLoad AttributeLoad AttributeLoad NameLoad attr attr Num body Assign AttributeStore NameLoad attr Call NameLoad Call NameLoad AttributeLoad AttributeLoad NameLoad attr attr orelse Raise Call NameLoad BinOpMod Str Call NameLoad AttributeLoad AttributeLoad NameLoad attr attr Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr If AttributeLoad NameLoad attr body Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign AttributeStore AttributeLoad NameLoad attr attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr If AttributeLoad NameLoad attr body Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad AttributeLoad NameLoad attr attr decorator_list decorator_list ClassDef bases NameLoad NameLoad body FunctionDef arguments args NameParam NameParam defaults body Assign AttributeStore NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr ListLoad NameLoad Assign AttributeStore NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr ListComp Call Call NameLoad NameLoad comprehension NameStore Call NameLoad AttributeLoad NameLoad attr Return AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Assign NameStore SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Slice Num For NameStore Call NameLoad AttributeLoad NameLoad attr body Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr NameLoad Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Index NameLoad Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr Call AttributeLoad NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam defaults body For NameStore Call NameLoad AttributeLoad NameLoad attr body Assign AttributeStore SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Index NameLoad decorator_list FunctionDef arguments args NameParam defaults body For NameStore Call NameLoad AttributeLoad NameLoad attr body Expr Call AttributeLoad NameLoad attr SubscriptLoad AttributeLoad AttributeLoad NameLoad attr attr Index NameLoad AttributeLoad SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr decorator_list decorator_list ClassDef bases NameLoad body FunctionDef arguments args NameParam defaults vararg body Assign AttributeStore NameLoad attr Call NameLoad NameLoad Assign AttributeStore NameLoad attr NameLoad Return NameLoad decorator_list FunctionDef arguments args NameParam defaults body Assign NameStore AttributeLoad SubscriptLoad AttributeLoad NameLoad attr Index Num attr For NameStore AttributeLoad NameLoad attr body If CompareNotEq NameLoad AttributeLoad NameLoad attr body Raise Call NameLoad BinOpMod Str TupleLoad NameLoad AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr BinOpAdd TupleLoad AttributeLoad NameLoad attr NameLoad Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body For NameStore Call NameLoad AttributeLoad NameLoad attr body Expr Call AttributeLoad NameLoad attr SubscriptLoad AttributeLoad NameLoad attr Index NameLoad AttributeLoad SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body For NameStore Call NameLoad AttributeLoad NameLoad attr body Expr Call AttributeLoad NameLoad attr AttributeLoad SubscriptLoad AttributeLoad NameLoad attr Index NameLoad attr SubscriptLoad AttributeLoad NameLoad attr Index NameLoad decorator_list decorator_list ClassDef bases NameLoad body FunctionDef arguments args NameParam NameParam defaults body Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr Num decorator_list FunctionDef arguments args NameParam NameParam NameParam defaults body Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr NameLoad Assign AttributeStore NameLoad attr BoolOpOr AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr ListLoad NameLoad NameLoad Return NameLoad decorator_list FunctionDef arguments args NameParam defaults body Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr Assign NameStore BinOpAdd SubscriptLoad NameLoad Index AttributeLoad NameLoad attr SubscriptLoad NameLoad Index AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr SubscriptLoad NameLoad Index AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr BinOpAdd BinOpAdd SubscriptLoad NameLoad Slice AttributeLoad NameLoad attr TupleLoad NameLoad SubscriptLoad NameLoad Slice BinOpAdd AttributeLoad NameLoad attr Num Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Expr Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad AttributeLoad NameLoad attr attr keyword AttributeLoad NameLoad attr keyword AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults body Expr Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr keyword AttributeLoad NameLoad attr keyword AttributeLoad NameLoad attr keyword AttributeLoad AttributeLoad NameLoad attr attr keyword AttributeLoad AttributeLoad NameLoad attr attr decorator_list decorator_list FunctionDef arguments args NameParam defaults body Return Call Call NameLoad NameLoad decorator_list
N/A N/A numpy np N/A cudarray ca base Op Output SplitMixin Unary Flatten N/A Unary N/A setup N/A N/A self N/A N/A N/A shape N/A N/A self x shape N/A N/A self shape N/A N/A shape N/A 0 N/A N/A np prod N/A shape N/A 1 N/A N/A self array N/A N/A ca zeros N/A self shape N/A N/A self grad_array N/A N/A ca zeros N/A self shape N/A fprop N/A N/A self N/A N/A N/A N/A self array N/A N/A ca reshape N/A N/A self x array N/A self shape N/A bprop N/A N/A self N/A N/A N/A N/A N/A self x grad_array N/A N/A ca reshape N/A self grad_array N/A N/A self x shape N/A N/A Reshape N/A Unary N/A __init__ N/A N/A self newshape N/A N/A N/A N/A N/A isinstance newshape tuple N/A N/A N/A hasattr newshape __iter__ N/A N/A newshape N/A tuple newshape N/A N/A newshape N/A newshape N/A N/A N/A 0 N/A N/A N/A list newshape count -1 1 N/A N/A N/A ValueError N/A invalid newshape: %s newshape N/A N/A self newshape newshape N/A setup N/A N/A self N/A N/A N/A size N/A N/A N/A self x array size N/A newshape_size N/A N/A np prod N/A self newshape N/A N/A self shape N/A tuple N/A N/A N/A d -1 d N/A N/A size newshape_size N/A d N/A self newshape N/A N/A self array N/A N/A ca zeros N/A self shape N/A N/A self grad_array N/A N/A ca zeros N/A self shape N/A fprop N/A N/A self N/A N/A N/A N/A self array N/A N/A ca reshape N/A N/A self x array N/A self shape N/A bprop N/A N/A self N/A N/A N/A N/A N/A self x grad_array N/A N/A ca reshape N/A self grad_array N/A N/A self x shape N/A N/A Slices N/A Op SplitMixin N/A __init__ N/A N/A self splits N/A N/A N/A N/A self splits splits N/A N/A self outputs N/A N/A N/A Output self N/A _ N/A range N/A N/A len N/A self splits 1 N/A __call__ N/A N/A self x N/A N/A N/A N/A self x x N/A N/A self inputs N/A x N/A N/A self bpropable N/A x bpropable N/A output N/A self outputs N/A N/A N/A output bpropable N/A self bpropable N/A N/A self outputs N/A setup N/A N/A self N/A N/A N/A splits N/A N/A N/A 0 N/A self splits N/A N/A N/A N/A self x shape N/A 0 N/A N/A self slices N/A N/A i N/A range N/A N/A len splits 1 N/A N/A N/A N/A N/A self slices append N/A N/A splits N/A i N/A splits N/A N/A i 1 N/A N/A i N/A start end N/A enumerate N/A self slices N/A N/A shape N/A N/A N/A end start N/A N/A N/A self x shape N/A 1 N/A N/A N/A N/A self outputs N/A i shape shape N/A N/A N/A N/A self outputs N/A i array N/A N/A N/A self x array N/A start end N/A N/A self bpropable N/A N/A N/A N/A N/A self outputs N/A i grad_array N/A N/A ca zeros shape N/A fprop N/A N/A self N/A N/A N/A N/A i N/A start end N/A enumerate N/A self slices N/A N/A N/A N/A N/A self outputs N/A i array N/A N/A N/A self x array N/A start end N/A bprop N/A N/A self N/A N/A N/A N/A i N/A start end N/A enumerate N/A self slices N/A N/A N/A N/A ca copyto N/A N/A N/A self x grad_array N/A start end N/A N/A N/A self outputs N/A i grad_array N/A N/A Transpose N/A Unary N/A __init__ N/A N/A self contiguous N/A False N/A N/A N/A self contiguous contiguous N/A setup N/A N/A self N/A N/A N/A N/A N/A len N/A N/A self x shape 1 N/A N/A N/A self shape N/A N/A 1 N/A N/A self x shape N/A N/A N/A N/A len N/A N/A self x shape 2 N/A N/A N/A self shape N/A tuple N/A reversed N/A N/A self x shape N/A N/A N/A ValueError N/A invalid shape for transpose: %s N/A str N/A N/A self x shape N/A N/A self array N/A N/A ca zeros N/A self shape N/A N/A self grad_array N/A N/A ca zeros N/A self shape N/A fprop N/A N/A self N/A N/A N/A N/A self array N/A N/A ca transpose N/A N/A self x array N/A N/A self contiguous N/A N/A N/A self array N/A N/A ca ascontiguousarray N/A self array N/A bprop N/A N/A self N/A N/A N/A N/A N/A self x grad_array N/A N/A ca transpose N/A self grad_array N/A N/A self contiguous N/A N/A N/A self array N/A N/A ca ascontiguousarray N/A N/A self x grad_array N/A N/A VSplit N/A Op SplitMixin N/A __init__ N/A N/A self len_x N/A N/A N/A N/A self n_splits len_x N/A __call__ N/A N/A self x N/A N/A N/A N/A self x x N/A N/A self inputs N/A x N/A N/A self bpropable N/A x bpropable N/A N/A self outputs N/A N/A N/A Output self N/A i N/A range N/A self n_splits N/A N/A self outputs N/A setup N/A N/A self N/A N/A N/A shape N/A N/A N/A self x shape N/A 1 N/A i N/A range N/A self n_splits N/A N/A N/A N/A N/A self outputs N/A i shape shape N/A N/A N/A N/A self outputs N/A i array N/A N/A N/A self x array N/A i N/A N/A N/A N/A self outputs N/A i grad_array N/A N/A ca zeros shape N/A fprop N/A N/A self N/A N/A N/A i N/A range N/A self n_splits N/A N/A N/A N/A N/A self outputs N/A i array N/A N/A N/A self x array N/A i N/A bprop N/A N/A self N/A N/A N/A i N/A range N/A self n_splits N/A N/A N/A N/A ca copyto N/A N/A N/A self x grad_array N/A i N/A N/A N/A self outputs N/A i grad_array N/A N/A VStack N/A Op N/A __call__ N/A N/A self N/A xs N/A N/A N/A self n_sources N/A len xs N/A N/A self inputs xs N/A self N/A setup N/A N/A self N/A N/A N/A shape N/A N/A N/A self inputs N/A 0 shape N/A expr N/A self inputs N/A N/A N/A shape N/A expr shape N/A N/A N/A ValueError N/A shape mismatch: %s and %s N/A shape N/A expr shape N/A N/A self shape N/A N/A N/A self n_sources shape N/A N/A self array N/A N/A ca zeros N/A self shape N/A N/A self grad_array N/A N/A ca zeros N/A self shape N/A fprop N/A N/A self N/A N/A N/A i N/A range N/A self n_sources N/A N/A N/A N/A ca copyto N/A N/A self array N/A i N/A N/A N/A self inputs N/A i array N/A bprop N/A N/A self N/A N/A N/A i N/A range N/A self n_sources N/A N/A N/A N/A ca copyto N/A N/A N/A self inputs N/A i grad_array N/A N/A self grad_array N/A i N/A N/A Concatenate N/A Op N/A __init__ N/A N/A self axis N/A N/A N/A N/A self axis axis N/A N/A self a_size -1 N/A __call__ N/A N/A self a b N/A N/A N/A N/A self a a N/A N/A self b b N/A N/A self bpropable N/A N/A a bpropable N/A b bpropable N/A N/A self inputs N/A a b N/A self N/A setup N/A N/A self N/A N/A N/A a_shp N/A N/A self a shape N/A b_shp N/A N/A self b shape N/A concat_size N/A N/A a_shp N/A N/A self axis N/A b_shp N/A N/A self axis N/A N/A self a_size N/A a_shp N/A N/A self axis N/A N/A self shape N/A N/A N/A a_shp N/A N/A self axis N/A concat_size N/A a_shp N/A N/A N/A self axis 1 N/A N/A self array N/A N/A ca zeros N/A self shape N/A N/A self grad_array N/A N/A ca zeros N/A self shape N/A fprop N/A N/A self N/A N/A N/A N/A N/A N/A ca extra concatenate N/A N/A self a array N/A N/A self b array axis N/A self axis out N/A self array N/A bprop N/A N/A self N/A N/A N/A N/A N/A N/A ca extra split N/A self grad_array a_size N/A self a_size axis N/A self axis out_a N/A N/A self a grad_array out_b N/A N/A self b grad_array N/A N/A transpose N/A N/A x N/A N/A N/A N/A N/A Transpose x N/A