Searched refs:dev_set (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 168 DeviceSet dev_set; in RunGrappler() local
169 for (auto* d : device_mgr->ListDevices()) dev_set.AddDevice(d); in RunGrappler()
170 grappler::VirtualCluster cluster(&dev_set); in RunGrappler()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 741 const std::shared_ptr<DeviceSet> dev_set = device_set(); in InstantiateMultiDevice() local
746 options.input_devices, options.output_devices, *dev_set, arg_nodes, in InstantiateMultiDevice()
769 *dev_set, options.config_proto, &graph, &data->lib_def_, in InstantiateMultiDevice()
793 optimization_options.device_set = dev_set.get(); in InstantiateMultiDevice()
806 dev_set.get(), default_device, in InstantiateMultiDevice()
824 &data->lib_def_, *dev_set, cpu_device, &graph); in InstantiateMultiDevice()
858 PartitionFunctionGraph(*dev_set, std::move(graph), &subgraphs)); in InstantiateMultiDevice()
931 runner([this, &pair, dev_set, comp_data, unique_name, data_lib_def, in InstantiateMultiDevice()
936 dev_set->FindDeviceByName(target)->device_type(); in InstantiateMultiDevice()
|