Path Lines of Code scheduler/tests/jobrequester_test.py 48 scheduler/tests/query_and_cancel_test.py 100 test/__init__.py 1 test/dlc_tests/__init__.py 1 test/dlc_tests/benchmark/__init__.py 1 test/dlc_tests/benchmark/bai/__init__.py 1 test/dlc_tests/benchmark/bai/mxnet/training/mxnet_single.toml 41 test/dlc_tests/benchmark/bai/mxnet/training/test_performance_mxnet_training.py 13 test/dlc_tests/benchmark/bai/tensorflow/training/tensorflow_single.toml 46 test/dlc_tests/benchmark/bai/tensorflow/training/test_performance_tensorflow_training.py 13 test/dlc_tests/benchmark/ec2/__init__.py 1 test/dlc_tests/benchmark/ec2/mxnet/__init__.py 1 test/dlc_tests/benchmark/ec2/mxnet/test_performance_mxnet_inference.py 65 test/dlc_tests/benchmark/ec2/mxnet/test_performance_mxnet_training.py 48 test/dlc_tests/benchmark/ec2/pytorch/__init__.py 1 test/dlc_tests/benchmark/ec2/pytorch/inference/__init__.py 1 test/dlc_tests/benchmark/ec2/pytorch/inference/test_performance_pytorch_inference.py 62 test/dlc_tests/benchmark/ec2/pytorch/training/__init__.py 1 test/dlc_tests/benchmark/ec2/pytorch/training/test_performance_pytorch_training.py 104 test/dlc_tests/benchmark/ec2/tensorflow/__init__.py 1 test/dlc_tests/benchmark/ec2/tensorflow/inference/__init__.py 1 test/dlc_tests/benchmark/ec2/tensorflow/inference/test_performance_tensorflow_inference.py 95 test/dlc_tests/benchmark/ec2/tensorflow/training/__init__.py 1 test/dlc_tests/benchmark/ec2/tensorflow/training/test_performance_tensorflow_training.py 92 test/dlc_tests/benchmark/sagemaker/__init__.py 1 test/dlc_tests/benchmark/sagemaker/huggingface/__init__.py 1 test/dlc_tests/benchmark/sagemaker/huggingface/inference/__init__.py 1 test/dlc_tests/benchmark/sagemaker/huggingface/inference/resources/__init__.py 1 test/dlc_tests/benchmark/sagemaker/huggingface/inference/test_performance_huggingface_inference.py 3 test/dlc_tests/benchmark/sagemaker/mxnet/training/resources/mx_sm_benchmark.py 44 test/dlc_tests/benchmark/sagemaker/mxnet/training/resources/scripts/mxnet_imagenet_resnet50.py 360 test/dlc_tests/benchmark/sagemaker/mxnet/training/resources/scripts/smtrain-resnet50-imagenet.sh 2 test/dlc_tests/benchmark/sagemaker/mxnet/training/test_performance_mxnet_sm_training.py 90 test/dlc_tests/benchmark/sagemaker/tensorflow/training/__init__.py 1 test/dlc_tests/benchmark/sagemaker/tensorflow/training/resources/tensorflow1/singletrain_cpu.sh 10 test/dlc_tests/benchmark/sagemaker/tensorflow/training/resources/tensorflow1/singletrain_gpu.sh 6 test/dlc_tests/benchmark/sagemaker/tensorflow/training/resources/tensorflow2/singletrain_cpu.sh 10 test/dlc_tests/benchmark/sagemaker/tensorflow/training/resources/tensorflow2/singletrain_gpu.sh 5 test/dlc_tests/benchmark/sagemaker/tensorflow/training/resources/tensorflow2/singletrain_gpu_without_xla.sh 5 test/dlc_tests/benchmark/sagemaker/tensorflow/training/resources/tf_sm_benchmark.py 83 test/dlc_tests/benchmark/sagemaker/tensorflow/training/test_performance_optimized_sm_training.py 156 test/dlc_tests/benchmark/sagemaker/tensorflow/training/test_performance_tensorflow_sm_training.py 129 test/dlc_tests/conftest.py 564 test/dlc_tests/container_tests/__init__.py 1 test/dlc_tests/ec2/__init__.py 1 test/dlc_tests/ec2/mxnet/__init__.py 1 test/dlc_tests/ec2/mxnet/inference/__init__.py 1 test/dlc_tests/ec2/mxnet/inference/test_mxnet_inference.py 144 test/dlc_tests/ec2/mxnet/training/__init__.py 1 test/dlc_tests/ec2/mxnet/training/test_mxnet_training.py 118 test/dlc_tests/ec2/pytorch/__init__.py 1 test/dlc_tests/ec2/pytorch/inference/__init__.py 1 test/dlc_tests/ec2/pytorch/inference/test_pytorch_inference.py 118 test/dlc_tests/ec2/pytorch/training/__init__.py 1 test/dlc_tests/ec2/pytorch/training/test_pytorch_training.py 164 test/dlc_tests/ec2/tensorflow/__init__.py 1 test/dlc_tests/ec2/tensorflow/inference/__init__.py 1 test/dlc_tests/ec2/tensorflow/inference/test_tensorflow_inference.py 198 test/dlc_tests/ec2/tensorflow/training/__init__.py 1 test/dlc_tests/ec2/tensorflow/training/test_tensorflow_training.py 200 test/dlc_tests/ec2/test_curand.py 19 test/dlc_tests/ec2/test_smclarify.py 79 test/dlc_tests/ec2/test_smdebug.py 152 test/dlc_tests/ecs/__init__.py 1 test/dlc_tests/ecs/conftest.py 96 test/dlc_tests/ecs/mxnet/__init__.py 1 test/dlc_tests/ecs/mxnet/inference/__init__.py 1 test/dlc_tests/ecs/mxnet/inference/test_ecs_mxnet_inference.py 123 test/dlc_tests/ecs/mxnet/training/__init__.py 1 test/dlc_tests/ecs/mxnet/training/test_ecs_mxnet_training.py 72 test/dlc_tests/ecs/pytorch/__init__.py 1 test/dlc_tests/ecs/pytorch/inference/__init__.py 1 test/dlc_tests/ecs/pytorch/inference/test_ecs_pytorch_inference.py 91 test/dlc_tests/ecs/pytorch/training/__init__.py 1 test/dlc_tests/ecs/pytorch/training/test_ecs_pytorch_training.py 92 test/dlc_tests/ecs/tensorflow/__init__.py 1 test/dlc_tests/ecs/tensorflow/inference/__init__.py 1 test/dlc_tests/ecs/tensorflow/inference/test_ecs_tensorflow_inference.py 159 test/dlc_tests/ecs/tensorflow/training/__init__.py 1 test/dlc_tests/ecs/tensorflow/training/test_ecs_tensorflow_training.py 41 test/dlc_tests/eks/__init__.py 1 test/dlc_tests/eks/conftest.py 1 test/dlc_tests/eks/eks_manifest_templates/aws_access/secret.yaml 8 test/dlc_tests/eks/eks_manifest_templates/kubeflow/install_kubeflow.sh 60 test/dlc_tests/eks/eks_manifest_templates/kubeflow/install_kubeflow_custom_kfctl.sh 60 test/dlc_tests/eks/eks_manifest_templates/kubeflow/kfctl_aws.yaml 118 test/dlc_tests/eks/eks_manifest_templates/mxnet/inference/single_node_cpu_inference.yaml 53 test/dlc_tests/eks/eks_manifest_templates/mxnet/inference/single_node_gpu_inference.yaml 54 test/dlc_tests/eks/eks_manifest_templates/mxnet/inference/single_node_neuron_inference.yaml 60 test/dlc_tests/eks/eks_manifest_templates/mxnet/training/multi_node_gpu_training.yaml 52 test/dlc_tests/eks/eks_manifest_templates/mxnet/training/multi_node_training_horovod_mnist.yaml 61 test/dlc_tests/eks/eks_manifest_templates/pytorch/inference/single_node_cpu_inference.yaml 53 test/dlc_tests/eks/eks_manifest_templates/pytorch/inference/single_node_gpu_inference.yaml 54 test/dlc_tests/eks/eks_manifest_templates/pytorch/inference/single_node_neuron_inference.yaml 70 test/dlc_tests/eks/eks_manifest_templates/pytorch/training/multi_node_gpu_training.yaml 40 test/dlc_tests/eks/eks_manifest_templates/ssm/install_ssm.yaml 38 test/dlc_tests/eks/eks_manifest_templates/tensorflow/inference/single_node_cpu_inference.yaml 66 test/dlc_tests/eks/eks_manifest_templates/tensorflow/inference/single_node_gpu_inference.yaml 67 test/dlc_tests/eks/eks_manifest_templates/tensorflow/inference/single_node_neuron_inference.yaml 75 test/dlc_tests/eks/eks_manifest_templates/tensorflow/training/multi_node_gpu_training.yaml 57 test/dlc_tests/eks/eks_manifest_templates/training/single_node_training.yaml 27 test/dlc_tests/eks/mxnet/__init__.py 1 test/dlc_tests/eks/mxnet/inference/__init__.py 1 test/dlc_tests/eks/mxnet/inference/test_eks_mxnet_inference.py 116 test/dlc_tests/eks/mxnet/training/__init__.py 1 test/dlc_tests/eks/mxnet/training/test_eks_mxnet_multinode_training.py 122 test/dlc_tests/eks/mxnet/training/test_eks_mxnet_training.py 141 test/dlc_tests/eks/pytorch/__init__.py 1 test/dlc_tests/eks/pytorch/inference/__init__.py 1 test/dlc_tests/eks/pytorch/inference/test_eks_pytorch_inference.py 86 test/dlc_tests/eks/pytorch/training/__init__.py 1 test/dlc_tests/eks/pytorch/training/test_eks_pytorch_training.py 256 test/dlc_tests/eks/tensorflow/__init__.py 1 test/dlc_tests/eks/tensorflow/inference/__init__.py 1 test/dlc_tests/eks/tensorflow/inference/test_eks_tensorflow_inference.py 135 test/dlc_tests/eks/tensorflow/training/__init__.py 1 test/dlc_tests/eks/tensorflow/training/test_eks_tensorflow_multi_node_training.py 72 test/dlc_tests/eks/tensorflow/training/test_eks_tensorflow_training.py 41 test/dlc_tests/release_candidate_integration/test_sm_profiler.py 124 test/dlc_tests/sanity/__init__.py 1 test/dlc_tests/sanity/quick_checks/__init__.py 1 test/dlc_tests/sanity/quick_checks/test_dlc_developer_config.py 38 test/dlc_tests/sanity/quick_checks/test_generate_coverage_doc.py 35 test/dlc_tests/sanity/quick_checks/test_git_secrets.py 37 test/dlc_tests/sanity/test_binary_visibility.py 13 test/dlc_tests/sanity/test_canary_integration.py 28 test/dlc_tests/sanity/test_dlc_labels.py 116 test/dlc_tests/sanity/test_pre_release.py 505 test/dlc_tests/sanity/test_pt_s3.py 23 test/dlc_tests/sanity/test_safety_check.py 207 test/dlc_tests/sanity/test_safety_report_file.py 62 test/dlc_tests/sanity/test_security_check.py 100 test/dlc_tests/sanity/test_telemetry.py 147 test/dlc_tests/sanity/test_torchvision.py 42 test/dlc_tests/sanity/test_utility_installation.py 99 test/sagemaker_tests/__init__.py 34 test/sagemaker_tests/autogluon/__init__.py 1 test/sagemaker_tests/autogluon/inference/__init__.py 1 test/sagemaker_tests/autogluon/inference/conftest.py 219 test/sagemaker_tests/autogluon/inference/integration/__init__.py 5 test/sagemaker_tests/autogluon/inference/integration/local/__init__.py 1 test/sagemaker_tests/autogluon/inference/integration/local/test_serving.py 44 test/sagemaker_tests/autogluon/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/autogluon/inference/integration/sagemaker/test_serving.py 52 test/sagemaker_tests/autogluon/inference/integration/sagemaker/timeout.py 32 test/sagemaker_tests/autogluon/inference/resources/scripts/tabular_serve.py 21 test/sagemaker_tests/autogluon/inference/utils/__init__.py 9 test/sagemaker_tests/autogluon/inference/utils/image_utils.py 28 test/sagemaker_tests/autogluon/inference/utils/local_mode_utils.py 23 test/sagemaker_tests/autogluon/training/__init__.py 1 test/sagemaker_tests/autogluon/training/conftest.py 219 test/sagemaker_tests/autogluon/training/integration/__init__.py 31 test/sagemaker_tests/autogluon/training/integration/local/__init__.py 1 test/sagemaker_tests/autogluon/training/integration/local/ag_tools.py 8 test/sagemaker_tests/autogluon/training/integration/local/local_mode_utils.py 22 test/sagemaker_tests/autogluon/training/integration/local/test_single_machine_training.py 57 test/sagemaker_tests/autogluon/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/autogluon/training/integration/sagemaker/test_training.py 38 test/sagemaker_tests/autogluon/training/integration/sagemaker/timeout.py 43 test/sagemaker_tests/autogluon/training/resources/data/config/config.cpu.yaml 15 test/sagemaker_tests/autogluon/training/resources/data/config/config.gpu.yaml 7 test/sagemaker_tests/autogluon/training/resources/data/config/config.vision.gpu.yaml 8 test/sagemaker_tests/autogluon/training/resources/scripts/train_cv.py 46 test/sagemaker_tests/autogluon/training/resources/scripts/train_tab.py 61 test/sagemaker_tests/autogluon/training/utils/__init__.py 9 test/sagemaker_tests/autogluon/training/utils/image_utils.py 28 test/sagemaker_tests/autogluon/training/utils/local_mode_utils.py 23 test/sagemaker_tests/huggingface/__init__.py 1 test/sagemaker_tests/huggingface/inference/__init__.py 1 test/sagemaker_tests/huggingface/inference/conftest.py 256 test/sagemaker_tests/huggingface/inference/integration/__init__.py 40 test/sagemaker_tests/huggingface/inference/integration/local/__init__.py 1 test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py 37 test/sagemaker_tests/huggingface/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_hub_model.py 55 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuron_hosting.py 51 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_sm_trained_model.py 58 test/sagemaker_tests/huggingface/inference/integration/sagemaker/timeout.py 44 test/sagemaker_tests/huggingface/inference/resources/__init__.py 1 test/sagemaker_tests/huggingface/inference/resources/code/call_model_fn_once.py 14 test/sagemaker_tests/huggingface/inference/resources/code/neuron_inference.py 27 test/sagemaker_tests/huggingface/inference/utils/__init__.py 9 test/sagemaker_tests/huggingface/inference/utils/image_utils.py 28 test/sagemaker_tests/huggingface/inference/utils/local_mode_utils.py 23 test/sagemaker_tests/huggingface_pytorch/__init__.py 1 test/sagemaker_tests/huggingface_pytorch/training/__init__.py 1 test/sagemaker_tests/huggingface_pytorch/training/conftest.py 221 test/sagemaker_tests/huggingface_pytorch/training/integration/__init__.py 7 test/sagemaker_tests/huggingface_pytorch/training/integration/local/__init__.py 1 test/sagemaker_tests/huggingface_pytorch/training/integration/local/test_single_machine_training.py 21 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_smdp.py 94 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_smmp.py 80 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/timeout.py 43 test/sagemaker_tests/huggingface_pytorch/training/integration/utils.py 17 test/sagemaker_tests/huggingface_pytorch/training/resources/__init__.py 1 test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train.py 73 test/sagemaker_tests/huggingface_pytorch/training/utils/__init__.py 9 test/sagemaker_tests/huggingface_pytorch/training/utils/image_utils.py 28 test/sagemaker_tests/huggingface_pytorch/training/utils/local_mode_utils.py 23 test/sagemaker_tests/huggingface_tensorflow/__init__.py 1 test/sagemaker_tests/huggingface_tensorflow/training/__init__.py 1 test/sagemaker_tests/huggingface_tensorflow/training/integration/__init__.py 50 test/sagemaker_tests/huggingface_tensorflow/training/integration/conftest.py 146 test/sagemaker_tests/huggingface_tensorflow/training/integration/local/__init__.py 1 test/sagemaker_tests/huggingface_tensorflow/training/integration/local/test_placeholder.py 8 test/sagemaker_tests/huggingface_tensorflow/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/huggingface_tensorflow/training/integration/sagemaker/test_smdataparallel.py 52 test/sagemaker_tests/huggingface_tensorflow/training/integration/sagemaker/timeout.py 18 test/sagemaker_tests/huggingface_tensorflow/training/integration/utils.py 17 test/sagemaker_tests/huggingface_tensorflow/training/resources/__init__.py 1 test/sagemaker_tests/huggingface_tensorflow/training/resources/local_scripts/train.py 73 test/sagemaker_tests/huggingface_tensorflow/training/resources/scripts/train.py 119 test/sagemaker_tests/mxnet/__init__.py 1 test/sagemaker_tests/mxnet/inference/__init__.py 1 test/sagemaker_tests/mxnet/inference/conftest.py 164 test/sagemaker_tests/mxnet/inference/integration/__init__.py 65 test/sagemaker_tests/mxnet/inference/integration/local/__init__.py 1 test/sagemaker_tests/mxnet/inference/integration/local/local_mode_utils.py 22 test/sagemaker_tests/mxnet/inference/integration/local/test_default_model_fn.py 34 test/sagemaker_tests/mxnet/inference/integration/local/test_gluon_hosting.py 28 test/sagemaker_tests/mxnet/inference/integration/local/test_hosting.py 27 test/sagemaker_tests/mxnet/inference/integration/local/test_onnx.py 28 test/sagemaker_tests/mxnet/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_batch_transform.py 47 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_elastic_inference.py 51 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_gluonnlp.py 37 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_hosting.py 30 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_neuron_hosting.py 38 test/sagemaker_tests/mxnet/inference/integration/sagemaker/timeout.py 85 test/sagemaker_tests/mxnet/inference/resources/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/default_handlers/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/default_handlers/model/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/default_handlers/model/code/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/default_handlers/model/code/eia_module.py 14 test/sagemaker_tests/mxnet/inference/resources/default_handlers/model/code/empty_module.py 1 test/sagemaker_tests/mxnet/inference/resources/dummy_hosting/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/dummy_hosting/code/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/dummy_hosting/code/dummy_hosting_module.py 7 test/sagemaker_tests/mxnet/inference/resources/gluon_hosting/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/gluon_hosting/model/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/gluon_hosting/model/code/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/gluon_hosting/model/code/gluon.py 14 test/sagemaker_tests/mxnet/inference/resources/gluonnlp/bert.py 41 test/sagemaker_tests/mxnet/inference/resources/neuron_handlers/model/code/mnist-neuron.py 59 test/sagemaker_tests/mxnet/inference/resources/onnx/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/onnx/onnx_model/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/onnx/onnx_model/code/__init__.py 1 test/sagemaker_tests/mxnet/inference/resources/onnx/onnx_model/code/onnx_import.py 10 test/sagemaker_tests/mxnet/training/__init__.py 1 test/sagemaker_tests/mxnet/training/conftest.py 163 test/sagemaker_tests/mxnet/training/integration/__init__.py 60 test/sagemaker_tests/mxnet/training/integration/local/__init__.py 1 test/sagemaker_tests/mxnet/training/integration/local/local_mode_utils.py 22 test/sagemaker_tests/mxnet/training/integration/local/test_horovod.py 66 test/sagemaker_tests/mxnet/training/integration/local/test_keras_training.py 23 test/sagemaker_tests/mxnet/training/integration/local/test_linear_regression.py 24 test/sagemaker_tests/mxnet/training/integration/local/test_mnist_training.py 34 test/sagemaker_tests/mxnet/training/integration/local/test_onnx.py 18 test/sagemaker_tests/mxnet/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_dgl.py 28 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_gluonnlp.py 26 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_horovod.py 38 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_training.py 40 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_training_smdebug.py 37 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_tuning.py 44 test/sagemaker_tests/mxnet/training/integration/sagemaker/timeout.py 18 test/sagemaker_tests/mxnet/training/integration/utils.py 23 test/sagemaker_tests/mxnet/training/resources/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/dgl_gcn/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/dgl_gcn/gcn.py 141 test/sagemaker_tests/mxnet/training/resources/hvdbasic/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/hvdbasic/train_hvd_basic.py 8 test/sagemaker_tests/mxnet/training/resources/keras/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/keras/keras_mnist.py 75 test/sagemaker_tests/mxnet/training/resources/linear_regression/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/linear_regression/linear_regression.py 59 test/sagemaker_tests/mxnet/training/resources/mnist/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/mnist/horovod_mnist.py 140 test/sagemaker_tests/mxnet/training/resources/mnist/mnist.py 96 test/sagemaker_tests/mxnet/training/resources/mnist/mnist_gluon_basic_hook_demo.py 164 test/sagemaker_tests/mxnet/training/resources/nlp/word_embedding.py 75 test/sagemaker_tests/mxnet/training/resources/onnx/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/onnx/code/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/onnx/code/onnx_export.py 31 test/sagemaker_tests/pytorch/__init__.py 16 test/sagemaker_tests/pytorch/inference/__init__.py 1 test/sagemaker_tests/pytorch/inference/conftest.py 243 test/sagemaker_tests/pytorch/inference/integration/__init__.py 30 test/sagemaker_tests/pytorch/inference/integration/local/__init__.py 1 test/sagemaker_tests/pytorch/inference/integration/local/test_serving.py 110 test/sagemaker_tests/pytorch/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/pytorch/inference/integration/sagemaker/test_mnist.py 84 test/sagemaker_tests/pytorch/inference/integration/sagemaker/test_neuron_hosting.py 65 test/sagemaker_tests/pytorch/inference/integration/sagemaker/timeout.py 44 test/sagemaker_tests/pytorch/inference/resources/__init__.py 1 test/sagemaker_tests/pytorch/inference/resources/code/call_model_fn_once.py 14 test/sagemaker_tests/pytorch/inference/resources/mnist/__init__.py 1 test/sagemaker_tests/pytorch/inference/resources/mnist/mnist.py 35 test/sagemaker_tests/pytorch/inference/resources/mnist/model_cpu/1d/__init__.py 1 test/sagemaker_tests/pytorch/inference/resources/mnist/model_cpu/1d/code/mnist_1d.py 29 test/sagemaker_tests/pytorch/inference/resources/mnist/model_cpu/__init__.py 1 test/sagemaker_tests/pytorch/inference/resources/mnist/model_cpu/code/call_model_fn_once.py 14 test/sagemaker_tests/pytorch/inference/resources/mnist/model_cpu/code/mnist.py 35 test/sagemaker_tests/pytorch/inference/resources/mnist/model_eia/mnist.py 33 test/sagemaker_tests/pytorch/inference/resources/mnist/model_gpu/__init__.py 1 test/sagemaker_tests/pytorch/inference/resources/mnist/model_gpu/code/mnist.py 35 test/sagemaker_tests/pytorch/inference/resources/resnet/model_neuron/code/resnet18.py 32 test/sagemaker_tests/pytorch/inference/utils/__init__.py 9 test/sagemaker_tests/pytorch/inference/utils/image_utils.py 28 test/sagemaker_tests/pytorch/inference/utils/local_mode_utils.py 23 test/sagemaker_tests/pytorch/training/__init__.py 1 test/sagemaker_tests/pytorch/training/conftest.py 246 test/sagemaker_tests/pytorch/training/integration/__init__.py 44 test/sagemaker_tests/pytorch/training/integration/local/__init__.py 1 test/sagemaker_tests/pytorch/training/integration/local/test_distributed_training.py 86 test/sagemaker_tests/pytorch/training/integration/local/test_single_machine_training.py 42 test/sagemaker_tests/pytorch/training/integration/sagemaker/__init__.py 51 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_dgl.py 55 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_distributed_operations.py 307 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_mnist.py 47 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_pt_s3_plugin.py 57 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_smdataparallel.py 127 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_training_smdebug.py 33 test/sagemaker_tests/pytorch/training/integration/sagemaker/timeout.py 43 test/sagemaker_tests/pytorch/training/resources/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/call_model_fn_once.py 14 test/sagemaker_tests/pytorch/training/resources/dgl-gcn/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/dgl-gcn/gcn.py 151 test/sagemaker_tests/pytorch/training/resources/distributed_operations.py 197 test/sagemaker_tests/pytorch/training/resources/efa/test_efa.sh 6 test/sagemaker_tests/pytorch/training/resources/fastai/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/fastai/cifar/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/fastai/cifar/cifar.py 26 test/sagemaker_tests/pytorch/training/resources/fastai/cifar/train_cifar.py 2 test/sagemaker_tests/pytorch/training/resources/fastai/mnist.py 13 test/sagemaker_tests/pytorch/training/resources/gpt2/data_pipeline.py 142 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/__init__.py 14 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/fp16.py 560 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/fp16util.py 251 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/loss_scaler.py 123 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/megatron/clip_grads.py 77 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/megatron/fp16.py 465 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/megatron/grad_scaler.py 35 test/sagemaker_tests/pytorch/training/resources/gpt2/learning_rates.py 89 test/sagemaker_tests/pytorch/training/resources/gpt2/train_gpt_simple.py 1014 test/sagemaker_tests/pytorch/training/resources/mnist/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/mnist/mnist.py 202 test/sagemaker_tests/pytorch/training/resources/mnist/mnist_1d.py 29 test/sagemaker_tests/pytorch/training/resources/mnist/smdataparallel_mnist.py 176 test/sagemaker_tests/pytorch/training/resources/mnist/smdataparallel_mnist_script_mode.sh 2 test/sagemaker_tests/pytorch/training/resources/mnist/smdataparallel_smmodelparallel_mnist_script_mode.sh 3 test/sagemaker_tests/pytorch/training/resources/mnist/smdebug_mnist.py 179 test/sagemaker_tests/pytorch/training/resources/mnist/smmodelparallel_mnist_script_mode.sh 2 test/sagemaker_tests/pytorch/training/resources/mnist/smmodelparallel_pt_mnist.py 256 test/sagemaker_tests/pytorch/training/resources/resnet18/main.py 285 test/sagemaker_tests/pytorch/training/resources/smdataparallel/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/smdataparallel/smdataparallel_throughput.py 170 test/sagemaker_tests/pytorch/training/utils/__init__.py 8 test/sagemaker_tests/pytorch/training/utils/image_utils.py 28 test/sagemaker_tests/pytorch/training/utils/local_mode_utils.py 23 test/sagemaker_tests/tensorflow/__init__.py 1 test/sagemaker_tests/tensorflow/inference/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/conftest.py 1 test/sagemaker_tests/tensorflow/inference/test/integration/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/integration/local/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/integration/local/conftest.py 74 test/sagemaker_tests/tensorflow/inference/test/integration/local/multi_model_endpoint_test_utils.py 31 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_container.py 221 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_multi_model_endpoint.py 193 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_multi_tfs.py 70 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_pre_post_processing.py 95 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_pre_post_processing_mme.py 124 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_tfs_batching.py 50 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/conftest.py 187 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_ei.py 50 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_tfs.py 108 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/util.py 177 test/sagemaker_tests/tensorflow/inference/test/perf/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/perf/ab.sh 14 test/sagemaker_tests/tensorflow/inference/test/perf/create-endpoint.sh 16 test/sagemaker_tests/tensorflow/inference/test/perf/create-model.sh 19 test/sagemaker_tests/tensorflow/inference/test/perf/data_generator.py 89 test/sagemaker_tests/tensorflow/inference/test/perf/delete-endpoint.sh 3 test/sagemaker_tests/tensorflow/inference/test/perf/ec2-perftest.sh 30 test/sagemaker_tests/tensorflow/inference/test/perf/perftest_endpoint.py 90 test/sagemaker_tests/tensorflow/inference/test/resources/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme/cifar/1540855709/saved_model.pb 1564 test/sagemaker_tests/tensorflow/inference/test/resources/mme/half_plus_three/00000123/saved_model.pb 531 test/sagemaker_tests/tensorflow/inference/test/resources/mme/half_plus_two/00000123/saved_model.pb 553 test/sagemaker_tests/tensorflow/inference/test/resources/mme_universal_script/code/inference.py 22 test/sagemaker_tests/tensorflow/inference/test/resources/mme_universal_script/half_plus_three/model/half_plus_three/00000123/saved_model.pb 531 test/sagemaker_tests/tensorflow/inference/test/resources/mme_universal_script/half_plus_three/model/half_plus_three/00000124/saved_model.pb 531 test/sagemaker_tests/tensorflow/inference/test/resources/models/half_plus_three/00000123/saved_model.pb 531 test/sagemaker_tests/tensorflow/inference/test/resources/models/half_plus_three/00000124/saved_model.pb 531 test/sagemaker_tests/tensorflow/tensorflow1_training/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/__init__.py 70 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/conftest.py 151 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/local/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/local/test_horovod.py 86 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/local/test_keras.py 34 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/local/test_training.py 133 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_experiments.py 72 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_horovod.py 66 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_mnist.py 180 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_tuning_model_dir.py 31 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/timeout.py 18 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/utils.py 17 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/gpu_device_placement.py 11 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/hvdbasic/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/hvdbasic/train_hvd_basic.py 8 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/hvdbasic/train_hvd_env_vars.py 13 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/keras_inception.py 16 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/mnist/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/mnist/horovod_mnist.py 81 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/mnist/mnist.py 47 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/mnist/mnist_estimator.py 102 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/mnist/tf1x_mnist_smdebug.py 78 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/test_dir_correct_model/12345/saved_model.pb 1 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/test_dir_wrong_parent_dir/not-digit/saved_model.pb 1 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/test_py_version/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/test_py_version/entry.py 6 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/tuning_model_dir/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/tuning_model_dir/entry.py 9 test/sagemaker_tests/tensorflow/tensorflow2_training/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/__init__.py 74 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/conftest.py 151 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/test_horovod.py 84 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/test_keras.py 34 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/test_training.py 134 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/recordio_utils.py 77 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_experiments.py 71 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_horovod.py 75 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_mnist.py 244 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_pipemode.py 85 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smdataparallel.py 120 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smmodelparallel.py 164 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_tuning_model_dir.py 31 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/timeout.py 18 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/utils.py 17 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/gpu_device_placement.py 11 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/hvdbasic/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/hvdbasic/train_hvd_basic.py 8 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/hvdbasic/train_hvd_env_vars.py 13 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/keras_inception.py 16 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/horovod_mnist.py 50 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist.py 38 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist_custom.py 121 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist_estimator.py 104 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist_smdebug.py 78 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/smdataparallel_mnist.py 49 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/smdataparallel_mnist_script_mode.sh 2 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/pipemode/pipemode.py 78 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smdataparallel/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smdataparallel/smdataparallel_throughput.py 141 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/send_receive_checkpoint.py 32 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/send_receive_checkpoint.sh 2 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/smdataparallel_mnist.py 49 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/smdataparallel_smmodelparallel_mnist_script_mode.sh 3 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/smmodelparallel_hvd2_conv.py 146 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/smmodelparallel_hvd2_conv.sh 2 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/smmodelparallel_hvd2_conv_multinode.py 146 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/smmodelparallel_tf2conv.sh 3 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/tf2_checkpoint_test.py 95 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/tf2_checkpoint_test.sh 2 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/tf2_conv.py 179 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/tf2_conv_xla.py 179 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smmodelparallel/utils.py 8 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/test_dir_correct_model/12345/saved_model.pb 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/test_dir_wrong_parent_dir/not-digit/saved_model.pb 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/test_py_version/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/test_py_version/entry.py 6 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/tuning_model_dir/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/tuning_model_dir/entry.py 9 test/test_utils/__init__.py 833 test/test_utils/benchmark.py 53 test/test_utils/ec2.py 424 test/test_utils/ecr.py 106 test/test_utils/ecs.py 504 test/test_utils/eks.py 238 test/test_utils/metrics.py 30 test/test_utils/pytest_cache.py 117 test/test_utils/sagemaker.py 267 test/test_utils/security.py 97 test/test_utils/test_reporting.py 193 test/testrunner.py 307