It looks like you are never using dist, so deleting it won’t affect the computation graph.
However, you could also just remove the computation in the first place, if you don’t need it for further computation.