from tensorflow.python.client import session


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

