Home
last modified time | relevance | path

Searched refs:use_multiple_devices (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Drpcbench_test.cc125 bool use_multiple_devices, const Cluster* cluster) { in CreateGraphDef() argument
142 s.WithDevice(cluster->devices[use_multiple_devices ? j : 0].name()), in CreateGraphDef()
170 bool use_multiple_devices) { in BM_Helper() argument
177 use_multiple_devices, cluster); in BM_Helper()
187 use_multiple_devices ? "Multi device" : "Single device", in BM_Helper()
/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.cc32 bool use_multiple_devices, bool insert_queue, in CreateGraphDef() argument
52 device_names[use_multiple_devices ? j % device_names.size() in CreateGraphDef()
59 device_names[use_multiple_devices ? j % device_names.size() in CreateGraphDef()