Path Lines of Code CHANGELOG.md 17 CODE_OF_CONDUCT.md 57 CONTRIBUTING.md 16 INSTALL.md 68 README.md 90 examples/cpp/AlexNet/alexnet.cc 313 examples/cpp/AlexNet/alexnet.cu 64 examples/cpp/AlexNet/alexnet.h 37 examples/cpp/DLRM/dlrm.cc 552 examples/cpp/DLRM/dlrm.cu 118 examples/cpp/DLRM/dlrm.h 66 examples/cpp/DLRM/preprocess_hdf.py 18 examples/cpp/DLRM/run_criteo_kaggle.sh 5 examples/cpp/DLRM/run_dlrm_kaggle_day1.sh 1 examples/cpp/DLRM/run_random.sh 4 examples/cpp/DLRM/run_summit.sh 6 examples/cpp/DLRM/run_summit_large.sh 6 examples/cpp/DLRM/strategies/dlrm_strategy.cc 275 examples/cpp/DLRM/strategies/dlrm_strategy.py 22 examples/cpp/DLRM/strategies/dlrm_strategy_16embs_16gpus.pb 22 examples/cpp/DLRM/strategies/dlrm_strategy_16embs_8gpus.pb 19 examples/cpp/DLRM/strategies/dlrm_strategy_8embs_8gpus.pb 11 examples/cpp/DLRM/strategies/dlrm_strategy_hetero.cc 49 examples/cpp/DLRM/strategies/gen_strategy.sh 15 examples/cpp/InceptionV3/inception.cc 403 examples/cpp/InceptionV3/inception.cu 64 examples/cpp/InceptionV3/inception.h 36 examples/cpp/ResNet/resnet.cc 366 examples/cpp/ResNet/resnet.cu 64 examples/cpp/ResNet/resnet.h 36 examples/cpp/Transformer/transformer.cc 291 examples/cpp/Transformer/transformer.cu 50 examples/cpp/Transformer/transformer.h 31 examples/cpp/candle_uno/candle_uno.cc 377 examples/cpp/candle_uno/candle_uno.cu 41 examples/cpp/candle_uno/candle_uno.h 35 examples/cpp/mixture_of_experts/moe.cc 386 examples/cpp/mixture_of_experts/moe.cu 62 examples/cpp/mixture_of_experts/moe.h 36 examples/python/keras/__init__.py 1 examples/python/keras/accuracy.py 7 examples/python/keras/callback.py 42 examples/python/keras/candle_uno/candle_uno.py 187 examples/python/keras/candle_uno/default_utils.py 458 examples/python/keras/candle_uno/file_utils.py 116 examples/python/keras/candle_uno/generic_utils.py 156 examples/python/keras/candle_uno/uno.py 216 examples/python/keras/candle_uno/uno_data.py 910 examples/python/keras/func_cifar10_alexnet.py 54 examples/python/keras/func_cifar10_cnn.py 41 examples/python/keras/func_cifar10_cnn_concat.py 52 examples/python/keras/func_cifar10_cnn_concat_model.py 53 examples/python/keras/func_cifar10_cnn_concat_seq_model.py 47 examples/python/keras/func_cifar10_cnn_nested.py 47 examples/python/keras/func_cifar10_cnn_net2net.py 85 examples/python/keras/func_mnist_cnn.py 47 examples/python/keras/func_mnist_cnn_concat.py 41 examples/python/keras/func_mnist_mlp.py 35 examples/python/keras/func_mnist_mlp_concat.py 55 examples/python/keras/func_mnist_mlp_concat2.py 60 examples/python/keras/func_mnist_mlp_net2net.py 56 examples/python/keras/reshape.py 38 examples/python/keras/seq_cifar10_cnn.py 34 examples/python/keras/seq_mnist_cnn.py 34 examples/python/keras/seq_mnist_cnn_nested.py 39 examples/python/keras/seq_mnist_cnn_net2net.py 66 examples/python/keras/seq_mnist_mlp.py 35 examples/python/keras/seq_mnist_mlp_net2net.py 62 examples/python/keras/seq_reuters_mlp.py 40 examples/python/keras/unary.py 39 examples/python/keras_exp/func_cifar10_cnn.py 38 examples/python/keras_exp/func_cifar10_cnn_concat.py 40 examples/python/keras_exp/func_cifar10_cnn_nested.py 44 examples/python/keras_exp/func_mnist_mlp.py 33 examples/python/keras_exp/func_mnist_mlp_concat.py 53 examples/python/native/__init__.py 1 examples/python/native/accuracy.py 7 examples/python/native/alexnet.py 98 examples/python/native/bert_proxy_native.py 83 examples/python/native/bert_proxy_run_script.sh 9 examples/python/native/cifar10_cnn.py 73 examples/python/native/cifar10_cnn_attach.py 88 examples/python/native/cifar10_cnn_concat.py 57 examples/python/native/dlrm.py 14 examples/python/native/inception.py 124 examples/python/native/mnist_cnn.py 63 examples/python/native/mnist_mlp.py 62 examples/python/native/mnist_mlp_attach.py 73 examples/python/native/multi_head_attention.py 63 examples/python/native/print_input.py 59 examples/python/native/print_layers.py 84 examples/python/native/resnet.py 87 examples/python/native/split.py 54 examples/python/native/tensor_attach.py 27 examples/python/onnx/accuracy.py 7 examples/python/onnx/alexnet.py 52 examples/python/onnx/alexnet_pt.py 38 examples/python/onnx/cifar10_cnn.py 52 examples/python/onnx/cifar10_cnn_keras.py 26 examples/python/onnx/cifar10_cnn_pt.py 37 examples/python/onnx/mnist_mlp.py 49 examples/python/onnx/mnist_mlp_keras.py 17 examples/python/onnx/mnist_mlp_pt.py 25 examples/python/onnx/resnet.py 53 examples/python/onnx/resnet_pt.py 239 examples/python/pytorch/cifar10_cnn.py 41 examples/python/pytorch/cifar10_cnn_torch.py 39 examples/python/pytorch/export_regnet_fx.py 6 examples/python/pytorch/mnist_mlp.py 36 examples/python/pytorch/mnist_mlp_torch.py 20 examples/python/pytorch/mnist_mlp_torch2.py 54 examples/python/pytorch/regnet.py 49 examples/python/pytorch/resnet.py 48 examples/python/pytorch/resnet152_DDP_training.py 97 examples/python/pytorch/resnet152_training.py 47 examples/python/pytorch/resnet_torch.py 233 examples/python/pytorch/torch_vision.py 47 examples/python/pytorch/torch_vision_torch.py 5 tests/ops/test_readme.md 7