from tensorflow.python.client import session

  def testDenseDistributed(self):
    worker, unused_ps = self._setupCluster()
    for dtype in [dtypes.float64, dtypes.half, dtypes.float32]:
      with session.Session(worker.target):
        var0, var1, update_op = self._setupDense(True, dtype)
        self._assertDenseCorrect(var0, var1, update_op)
