Path Lines of Code scheduler/tests/jobrequester_test.py 59 scheduler/tests/query_and_cancel_test.py 113 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 15 test/dlc_tests/benchmark/bai/tensorflow/training/tensorflow_single.toml 46 test/dlc_tests/benchmark/bai/tensorflow/training/test_performance_tensorflow_training.py 19 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 73 test/dlc_tests/benchmark/ec2/mxnet/test_performance_mxnet_training.py 50 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 173 test/dlc_tests/benchmark/ec2/pytorch/training/__init__.py 1 test/dlc_tests/benchmark/ec2/pytorch/training/test_performance_pytorch_training.py 172 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 117 test/dlc_tests/benchmark/ec2/tensorflow/training/__init__.py 1 test/dlc_tests/benchmark/ec2/tensorflow/training/test_performance_tensorflow_training.py 175 test/dlc_tests/benchmark/sagemaker/__init__.py 5 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 46 test/dlc_tests/benchmark/sagemaker/mxnet/training/resources/scripts/mxnet_imagenet_resnet50.py 432 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 113 test/dlc_tests/benchmark/sagemaker/pytorch/training/resources/inductor/run_inductor_huggingface.sh 36 test/dlc_tests/benchmark/sagemaker/pytorch/training/resources/inductor/run_inductor_timm.sh 34 test/dlc_tests/benchmark/sagemaker/pytorch/training/resources/inductor/run_inductor_torchbench.sh 47 test/dlc_tests/benchmark/sagemaker/pytorch/training/resources/inductor/test_inductor_helper.py 76 test/dlc_tests/benchmark/sagemaker/pytorch/training/test_performance_inductor.py 98 test/dlc_tests/benchmark/sagemaker/tensorflow/training/__init__.py 1 test/dlc_tests/benchmark/sagemaker/tensorflow/training/resources/clm/run_clm.py 423 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 95 test/dlc_tests/benchmark/sagemaker/tensorflow/training/test_performance_tensorflow_sm_training.py 147 test/dlc_tests/benchmark/sagemaker/tensorflow/training/test_trcomp_performance.py 256 test/dlc_tests/conftest.py 1371 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 211 test/dlc_tests/ec2/mxnet/training/__init__.py 1 test/dlc_tests/ec2/mxnet/training/test_mxnet_training.py 149 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 407 test/dlc_tests/ec2/pytorch/inference/test_pytorch_inference_inductor.py 131 test/dlc_tests/ec2/pytorch/training/__init__.py 1 test/dlc_tests/ec2/pytorch/training/common_cases.py 220 test/dlc_tests/ec2/pytorch/training/test_pytorch_training.py 615 test/dlc_tests/ec2/pytorch/training/test_pytorch_training_1_13.py 110 test/dlc_tests/ec2/pytorch/training/test_pytorch_training_2_1.py 111 test/dlc_tests/ec2/pytorch/training/test_pytorch_training_2_2.py 109 test/dlc_tests/ec2/pytorch/training/test_pytorch_training_2_3.py 109 test/dlc_tests/ec2/pytorch/training/test_pytorch_training_2_4.py 110 test/dlc_tests/ec2/pytorch/training/test_pytorch_training_2_5.py 109 test/dlc_tests/ec2/pytorch/training/test_pytorch_training_2_6.py 111 test/dlc_tests/ec2/smclarify_cases.py 48 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 431 test/dlc_tests/ec2/tensorflow/training/__init__.py 1 test/dlc_tests/ec2/tensorflow/training/test_tensorflow_training.py 361 test/dlc_tests/ec2/test_curand.py 23 test/dlc_tests/ec2/test_efa.py 251 test/dlc_tests/ec2/test_gdrcopy.py 35 test/dlc_tests/ec2/test_smclarify.py 46 test/dlc_tests/ec2/test_smdebug.py 214 test/dlc_tests/ec2/test_telemetry.py 527 test/dlc_tests/ec2/test_transformerengine.py 41 test/dlc_tests/ecs/__init__.py 1 test/dlc_tests/ecs/conftest.py 140 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 142 test/dlc_tests/ecs/mxnet/training/__init__.py 1 test/dlc_tests/ecs/mxnet/training/test_ecs_mxnet_training.py 95 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 196 test/dlc_tests/ecs/pytorch/training/__init__.py 1 test/dlc_tests/ecs/pytorch/training/test_ecs_pytorch_training.py 149 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 270 test/dlc_tests/ecs/tensorflow/training/__init__.py 1 test/dlc_tests/ecs/tensorflow/training/test_ecs_tensorflow_training.py 63 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 53 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 59 test/dlc_tests/eks/eks_manifest_templates/pytorch/inference/single_node_neuronx_inference.yaml 59 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 67 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 136 test/dlc_tests/eks/mxnet/training/__init__.py 1 test/dlc_tests/eks/mxnet/training/test_eks_mxnet_multinode_training.py 150 test/dlc_tests/eks/mxnet/training/test_eks_mxnet_training.py 146 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 140 test/dlc_tests/eks/pytorch/training/__init__.py 1 test/dlc_tests/eks/pytorch/training/test_eks_pytorch_training.py 195 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 226 test/dlc_tests/eks/tensorflow/training/__init__.py 1 test/dlc_tests/eks/tensorflow/training/test_eks_tensorflow_multi_node_training.py 89 test/dlc_tests/eks/tensorflow/training/test_eks_tensorflow_training.py 42 test/dlc_tests/release_candidate_integration/test_sm_profiler.py 151 test/dlc_tests/sanity/__init__.py 1 test/dlc_tests/sanity/quick_checks/__init__.py 1 test/dlc_tests/sanity/quick_checks/test_buildspecs.py 50 test/dlc_tests/sanity/quick_checks/test_dlc_developer_config.py 215 test/dlc_tests/sanity/quick_checks/test_generate_coverage_doc.py 47 test/dlc_tests/sanity/quick_checks/test_git_secrets.py 42 test/dlc_tests/sanity/quick_checks/test_prepare_dev_env.py 226 test/dlc_tests/sanity/quick_checks/test_required_fixture.py 85 test/dlc_tests/sanity/quick_checks/test_yml_config.py 112 test/dlc_tests/sanity/test_anaconda.py 29 test/dlc_tests/sanity/test_binary_visibility.py 29 test/dlc_tests/sanity/test_boottime_container_security.py 21 test/dlc_tests/sanity/test_canary_integration.py 31 test/dlc_tests/sanity/test_dlc_labels.py 215 test/dlc_tests/sanity/test_ecr_scan.py 316 test/dlc_tests/sanity/test_health_check.py 56 test/dlc_tests/sanity/test_mxnet_blas_mkl.py 32 test/dlc_tests/sanity/test_pre_release.py 874 test/dlc_tests/sanity/test_pt_s3.py 27 test/dlc_tests/sanity/test_remote_function_compatibility.py 30 test/dlc_tests/sanity/test_safety_check.py 1008 test/dlc_tests/sanity/test_safety_report_file.py 87 test/dlc_tests/sanity/test_scan_vulnerability_class_functionality.py 210 test/dlc_tests/sanity/test_telemetry.py 38 test/dlc_tests/sanity/test_torchvision.py 60 test/dlc_tests/sanity/test_utility_installation.py 195 test/sagemaker_tests/__init__.py 208 test/sagemaker_tests/autogluon/__init__.py 1 test/sagemaker_tests/autogluon/inference/__init__.py 1 test/sagemaker_tests/autogluon/inference/conftest.py 206 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 50 test/sagemaker_tests/autogluon/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/autogluon/inference/integration/sagemaker/test_serving.py 69 test/sagemaker_tests/autogluon/inference/integration/sagemaker/timeout.py 31 test/sagemaker_tests/autogluon/inference/resources/scripts/tabular_serve.py 25 test/sagemaker_tests/autogluon/inference/utils/__init__.py 9 test/sagemaker_tests/autogluon/inference/utils/image_utils.py 41 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 74 test/sagemaker_tests/autogluon/training/integration/local/__init__.py 1 test/sagemaker_tests/autogluon/training/integration/local/ag_tools.py 39 test/sagemaker_tests/autogluon/training/integration/local/local_mode_utils.py 22 test/sagemaker_tests/autogluon/training/integration/local/test_single_machine_training.py 88 test/sagemaker_tests/autogluon/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/autogluon/training/integration/sagemaker/test_training.py 48 test/sagemaker_tests/autogluon/training/integration/sagemaker/timeout.py 45 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 10 test/sagemaker_tests/autogluon/training/resources/scripts/train_cv.py 52 test/sagemaker_tests/autogluon/training/resources/scripts/train_tab.py 65 test/sagemaker_tests/autogluon/training/utils/__init__.py 9 test/sagemaker_tests/autogluon/training/utils/image_utils.py 41 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 267 test/sagemaker_tests/huggingface/inference/integration/__init__.py 58 test/sagemaker_tests/huggingface/inference/integration/local/__init__.py 1 test/sagemaker_tests/huggingface/inference/integration/local/test_serving.py 45 test/sagemaker_tests/huggingface/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_diffusers_model.py 110 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_hub_model.py 74 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_ipex_inference.py 77 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuron_hosting.py 78 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_decoder_hosting.py 110 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_encoder_hosting.py 110 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_no_context.py 98 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_sdxl_hosting.py 110 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_sm_trained_model.py 77 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_speech_model.py 90 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_torch_compile.py 105 test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_vision_model.py 83 test/sagemaker_tests/huggingface/inference/integration/sagemaker/timeout.py 45 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/diffusers_cpu_inference.py 25 test/sagemaker_tests/huggingface/inference/resources/code/diffusers_gpu_inference.py 27 test/sagemaker_tests/huggingface/inference/resources/code/ipex_inference.py 29 test/sagemaker_tests/huggingface/inference/resources/code/neuron_inference.py 29 test/sagemaker_tests/huggingface/inference/resources/code/neuronx_inference_decoder.py 21 test/sagemaker_tests/huggingface/inference/resources/code/neuronx_inference_encoder.py 11 test/sagemaker_tests/huggingface/inference/resources/code/neuronx_inference_sdxl.py 10 test/sagemaker_tests/huggingface/inference/resources/code/torch_compile_inference.py 31 test/sagemaker_tests/huggingface/inference/utils/__init__.py 9 test/sagemaker_tests/huggingface/inference/utils/image_utils.py 41 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 316 test/sagemaker_tests/huggingface_pytorch/training/integration/__init__.py 44 test/sagemaker_tests/huggingface_pytorch/training/integration/local/__init__.py 1 test/sagemaker_tests/huggingface_pytorch/training/integration/local/test_single_machine_training.py 61 test/sagemaker_tests/huggingface_pytorch/training/integration/local/test_trcomp.py 38 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_diffusers.py 43 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_neuronx.py 107 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_peft_sft.py 47 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_smdp.py 113 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_smmp.py 113 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_trcomp.py 365 test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/timeout.py 45 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/diffusers_entrypoint.py 37 test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train.py 80 test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train_peft_sft.py 108 test/sagemaker_tests/huggingface_pytorch/training/resources/scripts/train_torch_compiled.py 82 test/sagemaker_tests/huggingface_pytorch/training/utils/__init__.py 9 test/sagemaker_tests/huggingface_pytorch/training/utils/image_utils.py 41 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 54 test/sagemaker_tests/huggingface_tensorflow/training/integration/conftest.py 188 test/sagemaker_tests/huggingface_tensorflow/training/integration/local/__init__.py 1 test/sagemaker_tests/huggingface_tensorflow/training/integration/local/test_placeholder.py 9 test/sagemaker_tests/huggingface_tensorflow/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/huggingface_tensorflow/training/integration/sagemaker/test_smdataparallel.py 80 test/sagemaker_tests/huggingface_tensorflow/training/integration/sagemaker/test_trcomp.py 92 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 75 test/sagemaker_tests/huggingface_tensorflow/training/resources/scripts/train.py 128 test/sagemaker_tests/mxnet/__init__.py 1 test/sagemaker_tests/mxnet/inference/__init__.py 1 test/sagemaker_tests/mxnet/inference/conftest.py 185 test/sagemaker_tests/mxnet/inference/integration/__init__.py 19 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 38 test/sagemaker_tests/mxnet/inference/integration/local/test_gluon_hosting.py 33 test/sagemaker_tests/mxnet/inference/integration/local/test_hosting.py 30 test/sagemaker_tests/mxnet/inference/integration/local/test_onnx.py 31 test/sagemaker_tests/mxnet/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_batch_transform.py 59 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_elastic_inference.py 70 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_gluonnlp.py 45 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_hosting.py 36 test/sagemaker_tests/mxnet/inference/integration/sagemaker/test_neuron_hosting.py 42 test/sagemaker_tests/mxnet/inference/integration/sagemaker/timeout.py 99 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 36 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 191 test/sagemaker_tests/mxnet/training/integration/__init__.py 7 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 72 test/sagemaker_tests/mxnet/training/integration/local/test_keras_training.py 30 test/sagemaker_tests/mxnet/training/integration/local/test_linear_regression.py 33 test/sagemaker_tests/mxnet/training/integration/local/test_mnist_training.py 47 test/sagemaker_tests/mxnet/training/integration/local/test_onnx.py 25 test/sagemaker_tests/mxnet/training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_dgl.py 32 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_gluonnlp.py 30 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_horovod.py 48 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_training.py 51 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_training_smdebug.py 51 test/sagemaker_tests/mxnet/training/integration/sagemaker/test_tuning.py 53 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 129 test/sagemaker_tests/mxnet/training/resources/hvdbasic/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/hvdbasic/train_hvd_basic.py 10 test/sagemaker_tests/mxnet/training/resources/keras/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/keras/keras_mnist.py 81 test/sagemaker_tests/mxnet/training/resources/linear_regression/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/linear_regression/linear_regression.py 67 test/sagemaker_tests/mxnet/training/resources/mnist/__init__.py 1 test/sagemaker_tests/mxnet/training/resources/mnist/horovod_mnist.py 142 test/sagemaker_tests/mxnet/training/resources/mnist/mnist.py 118 test/sagemaker_tests/mxnet/training/resources/mnist/mnist_gluon_basic_hook_demo.py 176 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 96 test/sagemaker_tests/pytorch/inference/__init__.py 1 test/sagemaker_tests/pytorch/inference/conftest.py 242 test/sagemaker_tests/pytorch/inference/integration/__init__.py 42 test/sagemaker_tests/pytorch/inference/integration/local/__init__.py 1 test/sagemaker_tests/pytorch/inference/integration/local/test_serving.py 161 test/sagemaker_tests/pytorch/inference/integration/sagemaker/__init__.py 1 test/sagemaker_tests/pytorch/inference/integration/sagemaker/test_mnist.py 160 test/sagemaker_tests/pytorch/inference/integration/sagemaker/test_mnist_inductor.py 165 test/sagemaker_tests/pytorch/inference/integration/sagemaker/test_neuron_hosting.py 170 test/sagemaker_tests/pytorch/inference/integration/sagemaker/test_stabilityai.py 158 test/sagemaker_tests/pytorch/inference/integration/sagemaker/timeout.py 45 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 37 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 38 test/sagemaker_tests/pytorch/inference/resources/resnet/model_neuronx/code/resnet18.py 38 test/sagemaker_tests/pytorch/inference/resources/stabilityai/sdv2-1/model_gpu/code/sgm0.1.0/inference.py 144 test/sagemaker_tests/pytorch/inference/resources/stabilityai/sdxl-v1/model_gpu/code/sgm0.1.0/inference.py 226 test/sagemaker_tests/pytorch/inference/utils/__init__.py 9 test/sagemaker_tests/pytorch/inference/utils/image_utils.py 41 test/sagemaker_tests/pytorch/inference/utils/local_mode_utils.py 23 test/sagemaker_tests/pytorch/training/__init__.py 7 test/sagemaker_tests/pytorch/training/conftest.py 436 test/sagemaker_tests/pytorch/training/integration/__init__.py 102 test/sagemaker_tests/pytorch/training/integration/local/__init__.py 1 test/sagemaker_tests/pytorch/training/integration/local/test_distributed_training.py 88 test/sagemaker_tests/pytorch/training/integration/local/test_single_machine_training.py 58 test/sagemaker_tests/pytorch/training/integration/local/test_smppy_local.py 40 test/sagemaker_tests/pytorch/training/integration/sagemaker/__init__.py 133 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_dgl.py 83 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_dgl_inductor.py 57 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_distributed_operations.py 648 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_gdrcopy.py 51 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_mnist.py 80 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_mnist_inductor.py 96 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_neuron.py 196 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_pytorchddp.py 44 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_pytorchddp_inductor.py 45 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_smart_sifting.py 69 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_smdataparallel.py 215 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_smppy.py 120 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_torch_distributed.py 51 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_torch_distributed_inductor.py 44 test/sagemaker_tests/pytorch/training/integration/sagemaker/test_training_smdebug.py 73 test/sagemaker_tests/pytorch/training/integration/sagemaker/timeout.py 45 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/train.py 83 test/sagemaker_tests/pytorch/training/resources/dgl-gcn/train_dgl_lt_09x.py 128 test/sagemaker_tests/pytorch/training/resources/distributed_operations.py 234 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/mnist.py 9 test/sagemaker_tests/pytorch/training/resources/fastai/train_distributed.py 49 test/sagemaker_tests/pytorch/training/resources/gdrcopy/test_gdrcopy.sh 22 test/sagemaker_tests/pytorch/training/resources/gpt2/data_pipeline.py 161 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/__init__.py 14 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/fp16.py 587 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/fp16util.py 256 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/loss_scaler.py 126 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/megatron/clip_grads.py 84 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/megatron/fp16.py 477 test/sagemaker_tests/pytorch/training/resources/gpt2/fp16/megatron/grad_scaler.py 74 test/sagemaker_tests/pytorch/training/resources/gpt2/learning_rates.py 100 test/sagemaker_tests/pytorch/training/resources/gpt2/memory_tracker.py 94 test/sagemaker_tests/pytorch/training/resources/gpt2/train_gpt_simple.py 1039 test/sagemaker_tests/pytorch/training/resources/mnist/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/mnist/mnist.py 271 test/sagemaker_tests/pytorch/training/resources/mnist/mnist_1d.py 29 test/sagemaker_tests/pytorch/training/resources/mnist/pytorchddp_throughput_mnist.py 234 test/sagemaker_tests/pytorch/training/resources/mnist/smdataparallel_mnist.py 250 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 8 test/sagemaker_tests/pytorch/training/resources/mnist/smdebug_mnist.py 206 test/sagemaker_tests/pytorch/training/resources/mnist/smmodelparallel_mnist_script_mode.sh 2 test/sagemaker_tests/pytorch/training/resources/mnist/smmodelparallel_pt_mnist.py 260 test/sagemaker_tests/pytorch/training/resources/mnist/smppy_mnist.py 227 test/sagemaker_tests/pytorch/training/resources/mnist/torch_distributed_throughput_mnist.py 226 test/sagemaker_tests/pytorch/training/resources/neuron/all_reduce/all_reduce.py 42 test/sagemaker_tests/pytorch/training/resources/neuron/all_reduce/entrypoint.py 35 test/sagemaker_tests/pytorch/training/resources/neuron/mlp/entrypoint.py 38 test/sagemaker_tests/pytorch/training/resources/neuron/mlp/model.py 13 test/sagemaker_tests/pytorch/training/resources/neuron/mlp/train_torchrun.py 63 test/sagemaker_tests/pytorch/training/resources/smart_sifting/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/smart_sifting/train_plt_smart_sifting.py 229 test/sagemaker_tests/pytorch/training/resources/smdataparallel/__init__.py 1 test/sagemaker_tests/pytorch/training/resources/smdataparallel/smdataparallel_throughput.py 14 test/sagemaker_tests/pytorch/training/resources/smdataparallel/smdataparallel_throughput_post_ptbackend.py 122 test/sagemaker_tests/pytorch/training/resources/smdataparallel/smdataparallel_throughput_pre_ptbackend.py 149 test/sagemaker_tests/pytorch/training/utils/__init__.py 23 test/sagemaker_tests/pytorch/training/utils/image_utils.py 104 test/sagemaker_tests/pytorch/training/utils/local_mode_utils.py 23 test/sagemaker_tests/tensorflow/__init__.py 38 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/data/mme1/code/inference.py 24 test/sagemaker_tests/tensorflow/inference/test/data/mme2/code/inference.py 8 test/sagemaker_tests/tensorflow/inference/test/data/mme3/code/inference.py 28 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 90 test/sagemaker_tests/tensorflow/inference/test/integration/local/multi_model_endpoint_test_utils.py 29 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_container.py 233 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_multi_model_endpoint.py 182 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_multi_tfs.py 71 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_nginx_config.py 95 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_pre_post_processing.py 103 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_pre_post_processing_mme1.py 149 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_pre_post_processing_mme2.py 162 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_pre_post_processing_mme3.py 162 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_pre_post_processing_mme4.py 175 test/sagemaker_tests/tensorflow/inference/test/integration/local/test_tfs_batching.py 52 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/conftest.py 156 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_ei.py 73 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/test_tfs.py 367 test/sagemaker_tests/tensorflow/inference/test/integration/sagemaker/util.py 277 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 111 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 114 test/sagemaker_tests/tensorflow/inference/test/resources/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme1/code/inference.py 25 test/sagemaker_tests/tensorflow/inference/test/resources/mme2/code/inference.py 8 test/sagemaker_tests/tensorflow/inference/test/resources/mme2/models/half_plus_three/model/code/inference.py 23 test/sagemaker_tests/tensorflow/inference/test/resources/mme2/models/half_plus_two/model/code/inference.py 22 test/sagemaker_tests/tensorflow/inference/test/resources/mme3/code/inference.py 33 test/sagemaker_tests/tensorflow/inference/test/resources/mme3/code/lib/dummy_module/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme3/models/half_plus_three/model/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme3/models/half_plus_three/model/code/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme3/models/half_plus_three/model/code/inference.py 34 test/sagemaker_tests/tensorflow/inference/test/resources/mme3/models/half_plus_three/model/code/lib/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme3/models/half_plus_three/model/code/lib/dummy_module/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme4/models/half_plus_three/model/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme4/models/half_plus_three/model/code/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme4/models/half_plus_three/model/code/inference.py 36 test/sagemaker_tests/tensorflow/inference/test/resources/mme4/models/half_plus_three/model/code/lib/__init__.py 1 test/sagemaker_tests/tensorflow/inference/test/resources/mme4/models/half_plus_three/model/code/lib/dummy_module/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/__init__.py 17 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/conftest.py 169 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 37 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/local/test_training.py 147 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_experiments.py 80 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_horovod.py 83 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_mnist.py 226 test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_tuning_model_dir.py 44 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 10 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/hvdbasic/train_hvd_env_vars.py 13 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/keras_inception.py 17 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 97 test/sagemaker_tests/tensorflow/tensorflow1_training/resources/mnist/tf1x_mnist_smdebug.py 78 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 11 test/sagemaker_tests/tensorflow/tensorflow2_training/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/__init__.py 78 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/conftest.py 210 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/test_horovod.py 104 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/test_keras.py 37 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/test_smppy_local.py 34 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/local/test_training.py 152 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/recordio_utils.py 87 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_experiments.py 73 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_horovod.py 126 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_mnist.py 433 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_pipemode.py 111 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smdataparallel.py 177 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smmodelparallel.py 287 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smppy.py 139 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_smtrcomp.py 573 test/sagemaker_tests/tensorflow/tensorflow2_training/integration/sagemaker/test_tuning_model_dir.py 45 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 10 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/hvdbasic/train_hvd_env_vars.py 13 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/keras_inception.py 17 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mlm/run_mlm.py 464 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 42 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist_custom.py 130 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist_estimator.py 99 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist_smdebug.py 78 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/mnist_smppy.py 53 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/smdataparallel_mnist.py 51 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/mnist/smdataparallel_mnist_script_mode.sh 2 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/pipemode/pipemode.py 76 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smdataparallel/__init__.py 1 test/sagemaker_tests/tensorflow/tensorflow2_training/resources/smdataparallel/smdataparallel_throughput.py 129 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 51 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/smtrcomp/tensorflow_issue_58135.py 36 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 11 test/test_utils/__init__.py 1509 test/test_utils/benchmark.py 70 test/test_utils/ec2.py 1235 test/test_utils/ecr.py 224 test/test_utils/ecs.py 532 test/test_utils/eks.py 272 test/test_utils/imageutils.py 71 test/test_utils/metrics.py 38 test/test_utils/pytest_cache.py 171 test/test_utils/sagemaker.py 358 test/test_utils/security.py 869 test/test_utils/test_reporting.py 240 test/testrunner.py 444