Path	Lines of Code
config/buildspec_vanilla_framework_tests.yml	32
config/profiler/buildspec_profiler_sagemaker_pytorch_1_5_0_integration_tests.yml	12
config/profiler/buildspec_profiler_sagemaker_pytorch_1_6_0_integration_tests.yml	12
config/profiler/buildspec_profiler_sagemaker_pytorch_1_7_0_integration_tests.yml	12
config/profiler/buildspec_profiler_sagemaker_pytorch_1_8_0_integration_tests.yml	12
config/profiler/buildspec_profiler_sagemaker_tensorflow_2_2_1_integration_tests.yml	12
config/profiler/buildspec_profiler_sagemaker_tensorflow_2_3_1_integration_tests.yml	12
config/profiler/buildspec_profiler_sagemaker_tensorflow_2_4_1_integration_tests.yml	12
config/profiler/buildspec_profiler_sagemaker_tensorflow_2_5_0_integration_tests.yml	12
config/profiler/run_profiler_integration_tests.sh	80
tests/__init__.py	1
tests/analysis/__init__.py	1
tests/analysis/conftest.py	13
tests/analysis/exceptions/__init__.py	1
tests/analysis/exceptions/test_exceptions.py	72
tests/analysis/rules/__init__.py	1
tests/analysis/rules/test_rule_no_refresh.py	65
tests/analysis/tensors/__init__.py	1
tests/analysis/tensors/test_refresh.py	46
tests/analysis/trials/__init__.py	1
tests/analysis/trials/test_create.py	39
tests/analysis/trials/test_has_passed_step_scenarios.py	405
tests/analysis/trials/test_load_collections.py	18
tests/analysis/trials/test_local.py	25
tests/analysis/trials/test_modes.py	69
tests/analysis/trials/test_refresh.py	141
tests/analysis/trials/test_s3.py	65
tests/analysis/trials/test_tensors_api.py	79
tests/analysis/utils.py	69
tests/conftest.py	65
tests/constants.py	1
tests/core/__init__.py	1
tests/core/test_collections.py	62
tests/core/test_error_handling_agent.py	119
tests/core/test_handler.py	91
tests/core/test_index.py	48
tests/core/test_index_reader.py	35
tests/core/test_index_utils.py	58
tests/core/test_logger.py	22
tests/core/test_make_numpy_array.py	33
tests/core/test_modes.py	52
tests/core/test_numpy.py	37
tests/core/test_paths.py	51
tests/core/test_reduction_config.py	12
tests/core/test_save_config.py	18
tests/core/test_state_store.py	61
tests/core/test_timeline_writer.py	258
tests/core/test_training_end.py	33
tests/core/test_utils.py	229
tests/core/utils.py	108
tests/fixtures/__init__.py	1
tests/fixtures/error_handling_agent.py	12
tests/fixtures/tf2_training.py	91
tests/mxnet/__init__.py	1
tests/mxnet/mnist_gluon_model.py	128
tests/mxnet/test_custom_tensor.py	25
tests/mxnet/test_hook.py	64
tests/mxnet/test_hook_all_zero.py	45
tests/mxnet/test_hook_collections.py	45
tests/mxnet/test_hook_custom_collection.py	13
tests/mxnet/test_hook_loss_collection.py	47
tests/mxnet/test_hook_reduce_config.py	90
tests/mxnet/test_hook_save_all.py	35
tests/mxnet/test_hook_save_config.py	33
tests/mxnet/test_hook_save_scalar_mxnet.py	100
tests/mxnet/test_modes.py	50
tests/mxnet/test_spot_training.py	173
tests/mxnet/test_training_end.py	43
tests/profiler/__init__.py	1
tests/profiler/core/__init__.py	1
tests/profiler/core/test_algorithm_metric_readers.py	62
tests/profiler/core/test_horovodprofiler_events.py	53
tests/profiler/core/test_pandas_frames.py	227
tests/profiler/core/test_profiler_config_parser.py	351
tests/profiler/core/test_python_profiler.py	273
tests/profiler/core/test_smtfprofiler_events.py	39
tests/profiler/core/test_system_metric_reader.py	64
tests/profiler/core/test_system_profiler_file_parser.py	49
tests/profiler/core/test_tfprofiler_events.py	47
tests/profiler/core/test_utils.py	175
tests/profiler/core/utils.py	31
tests/profiler/pytorch/__init__.py	1
tests/profiler/pytorch/scripts/pytorch_rnn.py	30
tests/profiler/pytorch/test_pytorch_profiler.py	73
tests/profiler/pytorch/test_pytorch_profiler_rnn.py	56
tests/profiler/resources/__init__.py	1
tests/profiler/resources/profiler_config_parser_utils.py	161
tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.input_pipeline.pb	42
tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.kernel_stats.pb	1
tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.overview_page.pb	51
tests/profiler/resources/tfprofiler_timeline_traces/framework/tensorflow/detailed_profiling/2020080513/000000000/plugins/profile/2020_08_05_13_37_44/8c859046be41.ant.amazon.com.tensorflow_stats.pb	196
tests/profiler/tensorflow2/test_native_tf2_profiler.py	166
tests/profiler/tensorflow2/test_tf2_profiler.py	28
tests/profiler/tensorflow2/utils.py	14
tests/pytorch/__init__.py	1
tests/pytorch/test_collection.py	29
tests/pytorch/test_config_validator_pytorch.py	18
tests/pytorch/test_data_parallel.py	27
tests/pytorch/test_distributed_training.py	173
tests/pytorch/test_get_hook.py	38
tests/pytorch/test_hook_paths.py	27
tests/pytorch/test_hook_save_scalar_pytorch.py	119
tests/pytorch/test_loss.py	73
tests/pytorch/test_modes.py	105
tests/pytorch/test_no_name_clash.py	45
tests/pytorch/test_reduce_config.py	138
tests/pytorch/test_save_custom_tensor.py	25
tests/pytorch/test_simple_write.py	277
tests/pytorch/utils.py	60
tests/resources/mxnet/mnist_gluon_basic_hook_demo.py	140
tests/rules/__init__.py	1
tests/rules/action/__init__.py	1
tests/rules/action/test_message_action.py	55
tests/tensorflow/__init__.py	1
tests/tensorflow/hooks/__init__.py	1
tests/tensorflow/hooks/test_collection_defaults.py	92
tests/tensorflow/hooks/test_dist_horovod.py	15
tests/tensorflow/hooks/test_estimator_modes.py	210
tests/tensorflow/hooks/test_losses.py	11
tests/tensorflow/hooks/test_mirrored_strategy.py	338
tests/tensorflow/hooks/test_reductions.py	67
tests/tensorflow/hooks/test_save_all_full.py	23
tests/tensorflow/hooks/test_save_config.py	98
tests/tensorflow/hooks/test_session.py	51
tests/tensorflow/hooks/test_simple_include.py	125
tests/tensorflow/hooks/test_training_end.py	22
tests/tensorflow/hooks/test_weights_gradients.py	34
tests/tensorflow/hooks/test_write.py	78
tests/tensorflow/hooks/utils.py	46
tests/tensorflow/keras/__init__.py	1
tests/tensorflow/keras/test_keras.py	321
tests/tensorflow/keras/test_keras_lstm.py	76
tests/tensorflow/keras/test_keras_mirrored.py	453
tests/tensorflow/test_hook_save_scalar_tensorflow.py	138
tests/tensorflow/test_keras_to_estimator.py	34
tests/tensorflow/test_sagemaker.py	9
tests/tensorflow/test_tf_collections.py	32
tests/tensorflow/test_utils.py	193
tests/tensorflow/utils.py	5
tests/tensorflow2/__init__.py	1
tests/tensorflow2/test_concat_layer.py	33
tests/tensorflow2/test_config_validator_tensorflow.py	20
tests/tensorflow2/test_estimator.py	61
tests/tensorflow2/test_functional_model.py	46
tests/tensorflow2/test_get_hook.py	38
tests/tensorflow2/test_grad_tape_tf_function.py	65
tests/tensorflow2/test_keras.py	779
tests/tensorflow2/test_keras_lstm.py	78
tests/tensorflow2/test_keras_mirrored.py	397
tests/tensorflow2/test_mixed_precision_training.py	45
tests/tensorflow2/test_model_subclassing.py	70
tests/tensorflow2/test_sagemaker.py	9
tests/tensorflow2/test_should_save_tensor.py	53
tests/tensorflow2/test_support_dicts.py	37
tests/tensorflow2/test_tensorflow2_datatypes.py	23
tests/tensorflow2/utils.py	24
tests/utils.py	113
tests/xgboost/__init__.py	1
tests/xgboost/json_config.py	55
tests/xgboost/run_xgboost_model.py	18
tests/xgboost/test_hook.py	209
tests/xgboost/test_hook_save_scalar_xgboost.py	76
tests/xgboost/test_utils.py	38
tests/xgboost/test_xgboost_error_handling_agent.py	84
tests/zero_code_change/__init__.py	1
tests/zero_code_change/horovod_tests/__init__.py	1
tests/zero_code_change/horovod_tests/constants.py	6
tests/zero_code_change/horovod_tests/pytorch/__init__.py	1
tests/zero_code_change/horovod_tests/pytorch/test_hvd.py	143
tests/zero_code_change/horovod_tests/tensorflow/__init__.py	1
tests/zero_code_change/horovod_tests/tensorflow/test_estimator.py	85
tests/zero_code_change/horovod_tests/tensorflow/test_keras.py	76
tests/zero_code_change/horovod_tests/tensorflow2/test_keras_fit.py	132
tests/zero_code_change/horovod_tests/tf_utils.py	4
tests/zero_code_change/horovod_tests/utils.py	11
tests/zero_code_change/pt_utils.py	71
tests/zero_code_change/smdataparallel_tests/__init__.py	1
tests/zero_code_change/smdataparallel_tests/constants.py	5
tests/zero_code_change/smdataparallel_tests/pytorch/__init__.py	1
tests/zero_code_change/smdataparallel_tests/pytorch/test_smdataparallel.py	103
tests/zero_code_change/smdataparallel_tests/tensorflow2/__init__.py	1
tests/zero_code_change/smdataparallel_tests/tensorflow2/test_tf2_smdataparallel.py	115
tests/zero_code_change/smdataparallel_tests/utils.py	9
tests/zero_code_change/test_mxnet_error_handling_agent.py	98
tests/zero_code_change/test_mxnet_gluon_integration.py	121
tests/zero_code_change/test_pytorch_disengage_smdebug.py	21
tests/zero_code_change/test_pytorch_error_handling_agent.py	206
tests/zero_code_change/test_pytorch_integration.py	44
tests/zero_code_change/test_pytorch_multiprocessing.py	97
tests/zero_code_change/test_tensorflow2_bert.py	39
tests/zero_code_change/test_tensorflow2_error_handling_agent.py	230
tests/zero_code_change/test_tensorflow2_gradtape_integration.py	138
tests/zero_code_change/test_tensorflow2_integration.py	217
tests/zero_code_change/test_tensorflow_disengage_smdebug.py	20
tests/zero_code_change/test_tensorflow_error_handling_agent.py	96
tests/zero_code_change/test_tensorflow_integration.py	290
tests/zero_code_change/test_training_with_no_grad_updates.py	71
tests/zero_code_change/tf_utils.py	161
tests/zero_code_change/utils.py	32