MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / setUp

Method setUp

tensorflow/python/tpu/datasets_test.py:43–59  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

41class DatasetsTest(test.TestCase):
42
43 def setUp(self):
44 super(DatasetsTest, self).setUp()
45 self._coord = server_lib.Server.create_local_server()
46 self._worker = server_lib.Server.create_local_server()
47
48 self._cluster_def = cluster_pb2.ClusterDef()
49 worker_job = self._cluster_def.job.add()
50 worker_job.name = 'worker'
51 worker_job.tasks[0] = self._worker.target[len('grpc://'):]
52 coord_job = self._cluster_def.job.add()
53 coord_job.name = 'coordinator'
54 coord_job.tasks[0] = self._coord.target[len('grpc://'):]
55
56 session_config = config_pb2.ConfigProto(cluster_def=self._cluster_def)
57
58 self._sess = session.Session(self._worker.target, config=session_config)
59 self._worker_device = '/job:' + worker_job.name
60
61 def testTextLineDataset(self):
62 all_contents = []

Callers

nothing calls this directly

Calls 3

create_local_serverMethod · 0.80
addMethod · 0.45
SessionMethod · 0.45

Tested by

no test coverage detected