Home
last modified time | relevance | path

Searched refs:named_device (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/grappler/
Dcluster.i38 tensorflow::NamedDevice named_device; in _PyObjAs() local
39 if (!named_device.ParseFromString(string(c_string, py_size))) { in _PyObjAs()
45 if (out) *out = named_device; in _PyObjAs()
153 for (const auto& named_device : named_devices) { in TF_NewVirtualCluster() local
154 devices[named_device.name()]= named_device.properties(); in TF_NewVirtualCluster()
Dcluster_test.py103 named_device = device_properties_pb2.NamedDevice(
108 devices=[named_device])
113 estimated_perf = grappler_cluster.EstimatePerformance(named_device)
Dlayout_optimizer_test.py184 named_device = device_properties_pb2.NamedDevice()
185 named_device.name = '/GPU:0'
186 named_device.properties.type = 'GPU'
187 named_device.properties.num_cores = 24
188 named_device.properties.frequency = 1000
189 named_device.properties.environment['architecture'] = '4'
190 cluster = gcluster.Cluster(devices=[named_device])