Home
last modified time | relevance | path

Searched refs:gcluster (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/grappler/
Dgraph_placer_test.py110 gcluster = cluster.Cluster()
111 placed_mg = graph_placer.PlaceGraph(mg, allotted_time=15, cluster=gcluster)
117 available_devices = [device.name for device in gcluster.ListDevices()]
126 gcluster = GraphPlacerTest._buildCluster(num_gpus=1)
130 placed_mg = graph_placer.PlaceGraph(mg, allotted_time=15, cluster=gcluster)
134 available_devices = [device.name for device in gcluster.ListDevices()]
Dcost_analyzer.py23 from tensorflow.python.grappler import cluster as gcluster unknown
46 cluster = gcluster.Cluster(disable_detailed_stats=False)
69 cluster = gcluster.Cluster(
Dtf_optimizer.py25 from tensorflow.python.grappler import cluster as gcluster unknown
39 cluster = gcluster.Cluster()
Dcluster_test.py127 disable_detailed_stats=False, disable_timeline=False) as gcluster:
128 op_perfs, run_time, step_stats = gcluster.MeasureCosts(grappler_item)
134 with cluster.Provision() as gcluster:
135 op_names = gcluster.ListAvailableOps()
Dgraph_placer.py26 from tensorflow.python.grappler import cluster as gcluster unknown
54 cluster = gcluster.Cluster()
Dlayout_optimizer_test.py33 from tensorflow.python.grappler import cluster as gcluster unknown
190 cluster = gcluster.Cluster(devices=[named_device])