(This is one of my team member, and I think we had a couple of issues back then) We use something custom since this issue was raised, but I did not had time to investigate further . We also had some issues with tensorflow itself and it's been a while, sorry I can't help more