Description
Hi,
I was trying to use the "Color by device" option and my whole graph was gray. I learnt from this question that tensorboard only sees device assignments which are explicit, on the user side.
What would it take to show final device assignments, as they are when the graph is executed ? I realise this would have to be specified upfront, when actually saving the graph. Moreover, I believe that the main problem is that a graph is in no way bound to specific devices. This is what makes graphs and models shareable, in the first place.
But maybe the assignment could be written as a summary, much like all other stuff ?
Sorry if this issue is more on the tensorflow side than on the tensorboard, but it is unclear where implementation would have to start.