nmt/embed.cu
python/flexflow_dataloader.cu

nmt/softmax_data_parallel.cu
python/flexflow_dataloader.cu

nmt/lstm.cu
python/flexflow_dataloader.cu

nmt/rnn.cu
python/flexflow_dataloader.cu

nmt/linear.cu
python/flexflow_dataloader.cu