Path	Lines of Code
bootcamp_demo/ff_alexnet_cifar10.py	44
bootcamp_demo/keras_cnn_cifar10.py	36
bootcamp_demo/torch_alexnet_cifar10.py	37
cmake/cuda.cmake	46
cmake/cudnn.cmake	31
cmake/legion.cmake	22
cmake/nccl.cmake	18
cmake/utils.cmake	56
cmake/zlib.cmake	8
conda/meta.yaml	25
config/config.inc	71
flexflow/__init__.py	1
flexflow/core/__init__.py	33
flexflow/core/flexflow_cffi.py	1026
flexflow/core/flexflow_logger.py	29
flexflow/core/flexflow_pybind11.py	61
flexflow/core/flexflow_top.py	142
flexflow/core/flexflow_type.py	102
flexflow/keras/__init__.py	7
flexflow/keras/backend/__init__.py	11
flexflow/keras/backend/flexflow_backend.py	5
flexflow/keras/callbacks.py	58
flexflow/keras/datasets/__init__.py	4
flexflow/keras/datasets/cifar.py	19
flexflow/keras/datasets/cifar10.py	23
flexflow/keras/datasets/mnist.py	13
flexflow/keras/datasets/reuters.py	50
flexflow/keras/initializers.py	33
flexflow/keras/layers/__init__.py	7
flexflow/keras/layers/base_layer.py	122
flexflow/keras/layers/convolutional.py	139
flexflow/keras/layers/core.py	227
flexflow/keras/layers/input_layer.py	29
flexflow/keras/layers/merge.py	85
flexflow/keras/layers/normalization.py	42
flexflow/keras/layers/pool.py	86
flexflow/keras/losses.py	26
flexflow/keras/metrics.py	46
flexflow/keras/models/__init__.py	4
flexflow/keras/models/base_model.py	398
flexflow/keras/models/model.py	67
flexflow/keras/models/sequential.py	40
flexflow/keras/models/tensor.py	96
flexflow/keras/optimizers.py	31
flexflow/keras/preprocessing/__init__.py	7
flexflow/keras/preprocessing/sequence.py	9
flexflow/keras/preprocessing/text.py	9
flexflow/keras/utils/__init__.py	5
flexflow/keras/utils/data_utils.py	359
flexflow/keras/utils/generic_utils.py	335
flexflow/keras/utils/np_utils.py	22
flexflow/keras_exp/__init__.py	1
flexflow/keras_exp/models/__init__.py	2
flexflow/keras_exp/models/model.py	351
flexflow/keras_exp/models/tensor.py	62
flexflow/onnx/__init__.py	1
flexflow/onnx/model.py	292
flexflow/torch/__init__.py	1
flexflow/torch/fx.py	420
flexflow/torch/model.py	296
flexflow/torch/nn/__init__.py	1
flexflow/torch/nn/modules/__init__.py	7
flexflow/torch/nn/modules/module.py	17
include/accessor.h	51
include/config.h	135
include/cuda_helper.h	83
include/ffconst.h	128
include/initializer.h	78
include/loss_functions.h	29
include/mapper.h	347
include/metrics_functions.h	47
include/model.h	1734
include/optimizer.h	64
include/random_utils.h	33
include/recompile.h	22
include/simulator.h	339
include/tensor.h	53
nmt/embed.cu	349
nmt/linear.cu	516
nmt/lstm.cu	554
nmt/nmt.cc	272
nmt/ops.h	121
nmt/rnn.cu	680
nmt/rnn.h	249
nmt/rnn_mapper.cc	93
nmt/rnn_mapper.h	25
nmt/softmax_data_parallel.cu	361
python/bindings.cc	193
python/flexflow.py	77
python/flexflow/__init__.py	1
python/flexflow/core/__init__.py	33
python/flexflow/core/flexflow_cffi.py	1026
python/flexflow/core/flexflow_logger.py	29
python/flexflow/core/flexflow_pybind11.py	61
python/flexflow/core/flexflow_top.py	142
python/flexflow/core/flexflow_type.py	102
python/flexflow/keras/__init__.py	7
python/flexflow/keras/backend/__init__.py	11
python/flexflow/keras/backend/flexflow_backend.py	5
python/flexflow/keras/callbacks.py	58
python/flexflow/keras/datasets/__init__.py	4
python/flexflow/keras/datasets/cifar.py	19
python/flexflow/keras/datasets/cifar10.py	23
python/flexflow/keras/datasets/mnist.py	13
python/flexflow/keras/datasets/reuters.py	50
python/flexflow/keras/initializers.py	33
python/flexflow/keras/layers/__init__.py	7
python/flexflow/keras/layers/base_layer.py	122
python/flexflow/keras/layers/convolutional.py	139
python/flexflow/keras/layers/core.py	227
python/flexflow/keras/layers/input_layer.py	29
python/flexflow/keras/layers/merge.py	85
python/flexflow/keras/layers/normalization.py	42
python/flexflow/keras/layers/pool.py	86
python/flexflow/keras/losses.py	26
python/flexflow/keras/metrics.py	46
python/flexflow/keras/models/__init__.py	4
python/flexflow/keras/models/base_model.py	398
python/flexflow/keras/models/model.py	67
python/flexflow/keras/models/sequential.py	40
python/flexflow/keras/models/tensor.py	96
python/flexflow/keras/optimizers.py	31
python/flexflow/keras/preprocessing/__init__.py	7
python/flexflow/keras/preprocessing/sequence.py	9
python/flexflow/keras/preprocessing/text.py	9
python/flexflow/keras/utils/__init__.py	5
python/flexflow/keras/utils/data_utils.py	359
python/flexflow/keras/utils/generic_utils.py	335
python/flexflow/keras/utils/np_utils.py	22
python/flexflow/keras_exp/__init__.py	1
python/flexflow/keras_exp/models/__init__.py	2
python/flexflow/keras_exp/models/model.py	351
python/flexflow/keras_exp/models/tensor.py	62
python/flexflow/onnx/__init__.py	1
python/flexflow/onnx/model.py	292
python/flexflow/torch/__init__.py	1
python/flexflow/torch/fx.py	420
python/flexflow/torch/model.py	296
python/flexflow/torch/nn/__init__.py	1
python/flexflow/torch/nn/modules/__init__.py	7
python/flexflow/torch/nn/modules/module.py	17
python/flexflow_c.cc	1720
python/flexflow_c.h	674
python/flexflow_dataloader.cc	810
python/flexflow_dataloader.cu	144
python/flexflow_dataloader.h	123
python/legion_cffi.py.in	24
python/legion_cffi_build.py	34
python/main.cc	63
python/setup.py	26
scripts/cnn.h	493
scripts/simulator.cc	1314
src/loss_functions/loss_functions.cu	227
src/mapper/mapper.cc	1305
src/metrics_functions/metrics_functions.cc	60
src/metrics_functions/metrics_functions.cu	300
src/ops/aggregate.cu	574
src/ops/aggregate_spec.cu	565
src/ops/attention.cu	723
src/ops/batch_matmul.cu	633
src/ops/batch_norm.cu	648
src/ops/cache.cu	364
src/ops/concat.cu	478
src/ops/conv_2d.cu	1002
src/ops/dropout.cu	386
src/ops/element_binary.cu	840
src/ops/element_unary.cu	713
src/ops/embedding.cc	411
src/ops/embedding.cu	424
src/ops/flat.cu	284
src/ops/fused.cu	973
src/ops/group_by.cu	437
src/ops/linear.cu	1065
src/ops/pool_2d.cu	472
src/ops/reshape.cu	334
src/ops/reverse.cu	309
src/ops/softmax.cu	424
src/ops/split.cu	328
src/ops/topk.cu	688
src/ops/transpose.cu	415
src/recompile/recompile_state.cc	19
src/runtime/accessor_kernel.cu	156
src/runtime/cpp_driver.cc	19
src/runtime/cuda_helper.cu	238
src/runtime/initializer.cc	287
src/runtime/initializer_kernel.cu	262
src/runtime/machine_model.cc	748
src/runtime/model.cc	2768
src/runtime/model.cu	444
src/runtime/optimizer.cc	296
src/runtime/optimizer_kernel.cu	342
src/runtime/simulator.cc	534
src/runtime/simulator.cu	143
src/runtime/strategy.cc	151