Are you looking for a tool to visualize the computation graph (model architecture)?
If so, have a look at this post 9, where some tools were suggested.
If not, could you explain what you mean by flow a bit more, please?