Distributed Tensorflow training

Question

I am new to Tensorflow and trying to implement distributed Tensorflow using ParameterStrategy based on the Documentation. So far I have the code below

import mult