Replace all `keras` imports with `tensorflow.keras`.#570

copybara-service[bot] wants to merge 1 commit intomaingoogle/edward2:mainfrom

cl_559730941google/edward2:cl_559730941Copy head branch name to clipboard