smdebug/pytorch/hook.py
smdebug/tensorflow/session.py

smdebug/pytorch/collection.py
smdebug/tensorflow/collection.py

smdebug/pytorch/hook.py
smdebug/tensorflow/base_hook.py

smdebug/pytorch/hook.py
smdebug/tensorflow/keras.py