Imported from https://github.com/kibernetika-ai/re3-tensorflow