Module Import alias Import alias identifier Import alias ImportFrom alias ImportFrom alias alias ImportFrom alias Expr Call AttributeLoad NameLoad attr keyword Num Assign NameStore Call AttributeLoad NameLoad attr FunctionDef arguments args NameParam NameParam defaults body Assign TupleStore TupleStore NameStore NameStore TupleStore NameStore NameStore NameStore Call NameLoad keyword NameLoad Assign NameStore Call NameLoad NameLoad NameLoad keyword NameLoad Assign AttributeStore NameLoad attr AttributeLoad NameLoad attr For NameStore Call NameLoad Num body For TupleStore NameStore NameStore NameLoad body Print AttributeLoad NameLoad attr AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr Num decorator_list FunctionDef arguments args NameParam defaults body Assign NameStore Str Assign NameStore Str With Call NameLoad NameLoad Str NameStore body Expr Call AttributeLoad NameLoad attr NameLoad Assign AttributeStore AttributeLoad NameLoad attr attr Num Assign NameStore Num Assign NameStore Num Assign TupleStore NameStore NameStore Call AttributeLoad NameLoad attr NameLoad keyword NameLoad Assign NameStore Call NameLoad NameLoad NameLoad Assign NameStore Call NameLoad NameLoad NameLoad keyword AttributeLoad NameLoad attr Assign AttributeStore NameLoad attr AttributeLoad NameLoad attr Assign NameStore AttributeLoad AttributeLoad NameLoad attr attr For TupleStore NameStore TupleStore NameStore NameStore Call NameLoad NameLoad body If CompareGt NameLoad Num body Break Assign NameStore ListComp SubscriptLoad AttributeLoad NameLoad attr Index NameLoad comprehension NameStore Call AttributeLoad Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr keyword Num attr For NameStore Call NameLoad NameLoad body Assign NameStore Call AttributeLoad Str attr SubscriptLoad NameLoad Slice NameLoad NameLoad Assign NameStore BinOpAdd BinOpMult NameLoad NameLoad BinOpMult BinOpMult NameLoad NameLoad AttributeLoad NameLoad attr Assign NameStore SubscriptLoad NameLoad Slice NameLoad BinOpAdd NameLoad NameLoad Assert CompareEq NameLoad NameLoad Assign NameStore Call NameLoad BinOpMult Call NameLoad NameLoad BinOpSub Num NameLoad For TupleStore NameStore TupleStore NameStore NameStore Call NameLoad NameLoad body If CompareGt NameLoad Num body Break Assign NameStore ListComp SubscriptLoad AttributeLoad NameLoad attr Index NameLoad comprehension NameStore Call AttributeLoad Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr keyword Num attr For NameStore Call NameLoad NameLoad body Assign NameStore Call AttributeLoad Str attr SubscriptLoad NameLoad Slice NameLoad NameLoad Assign NameStore BinOpAdd BinOpAdd BinOpMult NameLoad NameLoad BinOpMult BinOpMult NameLoad NameLoad AttributeLoad NameLoad attr NameLoad Assign NameStore SubscriptLoad NameLoad Slice NameLoad BinOpAdd NameLoad NameLoad Assert CompareEq NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad Expr Call AttributeLoad NameLoad attr NameLoad Expr Call AttributeLoad NameLoad attr NameLoad decorator_list
N/A N/A logging N/A numpy np N/A os neon NervanaObject neon.data ArrayIterator load_mnist neon.data.text Text N/A N/A N/A logging basicConfig level 20 N/A logger N/A N/A logging getLogger test_dataset N/A N/A backend_default data N/A N/A N/A N/A N/A X_train y_train N/A X_test y_test nclass N/A load_mnist path data N/A train_set N/A ArrayIterator X_train y_train nclass nclass N/A N/A train_set be N/A NervanaObject be N/A i N/A range 2 N/A N/A N/A X_batch y_batch train_set N/A N/A N/A X_batch shape N/A y_batch shape N/A N/A train_set index 0 N/A test_text N/A N/A backend_default N/A N/A N/A text_data Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum. N/A data_path tmp_test_text_data N/A N/A open data_path w f N/A N/A N/A N/A f write text_data N/A N/A N/A NervanaObject be bsz 4 N/A time_steps 6 N/A valid_split 0.2 N/A N/A train_path valid_path N/A N/A Text create_valid_file data_path valid_split valid_split N/A train_set N/A Text time_steps train_path N/A valid_set N/A Text time_steps valid_path vocab N/A train_set vocab N/A N/A train_set be N/A NervanaObject be N/A bsz N/A N/A train_set be bsz N/A N/A i N/A X_batch y_batch N/A enumerate train_set N/A N/A N/A i 2 N/A N/A N/A chars N/A N/A N/A train_set index_to_token N/A x N/A x N/A N/A N/A N/A np argmax N/A N/A X_batch get axis 0 tolist N/A batch N/A range bsz N/A N/A sent N/A N/A join N/A chars N/A batch bsz N/A start N/A N/A i time_steps N/A N/A batch time_steps N/A train_set nbatches N/A sent_ref N/A text_data N/A start N/A start time_steps N/A N/A sent sent_ref N/A valid_start N/A int N/A N/A len text_data N/A 1 valid_split N/A N/A i N/A X_batch y_batch N/A enumerate valid_set N/A N/A N/A i 2 N/A N/A N/A chars N/A N/A N/A train_set index_to_token N/A x N/A x N/A N/A N/A N/A np argmax N/A N/A X_batch get axis 0 tolist N/A batch N/A range bsz N/A N/A sent N/A N/A join N/A chars N/A batch bsz N/A start N/A N/A N/A i time_steps N/A N/A batch time_steps N/A valid_set nbatches valid_start N/A sent_ref N/A text_data N/A start N/A start time_steps N/A N/A sent sent_ref N/A N/A N/A os remove data_path N/A N/A N/A os remove train_path N/A N/A N/A os remove valid_path N/A