Home
last modified time | relevance | path

Searched refs:device_set (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.cc125 DeviceSet device_set; in RunMain() local
127 device_set.AddDevice(device.get()); in RunMain()
129 options.device_set = &device_set; in RunMain()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc131 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet() argument
191 DeviceSet device_set; in TEST() local
195 device_set.AddDevice(d); in TEST()
197 cluster.SetDeviceSet(&device_set); in TEST()
Dconvert_graph.cc760 auto device_set = params.cluster->GetDeviceSet(); in GetDeviceAndAllocator() local
765 device_set->FindMatchingDevices(parsed_name, &devices); in GetDeviceAndAllocator()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster.cc372 std::unique_ptr<DeviceSet> device_set; in CreateSession() local
429 device_set.reset(new DeviceSet); in CreateSession()
431 device_set->AddDevice(d.get()); in CreateSession()
436 device_set->set_client_device(d.get()); in CreateSession()
447 device_set.reset(new DeviceSet); in CreateSession()
449 device_set->AddDevice(d.get()); in CreateSession()
453 device_set->AddDevice(d); in CreateSession()
456 device_set->set_client_device(d); in CreateSession()
462 CHECK(device_set->client_device()) << "No client device found. Missing " in CreateSession()
474 std::move(device_set), std::move(filtered_worker_list)); in CreateSession()
Dmaster_env.h87 std::unique_ptr<DeviceSet> device_set,
Dmaster_session.h54 std::unique_ptr<DeviceSet> device_set,
Dcollective_param_resolver_distributed.cc229 gr->device_set.insert(dn); in UpdateGroupCache()
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc26 const DeviceSet& device_set, std::unique_ptr<Graph> graph, in PartitionFunctionGraph() argument
41 [&device_set](const string& name) -> int64 { in PartitionFunctionGraph()
42 const Device* d = device_set.FindDeviceByName(name); in PartitionFunctionGraph()
Dcollective_param_resolver_local.cc81 if (!gr->device_set.empty()) { in CompleteGroupLocal()
98 auto it = gr->device_set.find(device); in CompleteGroupLocal()
99 if (it == gr->device_set.end()) { in CompleteGroupLocal()
100 if (gr->device_set.size() == gr->group.group_size) { in CompleteGroupLocal()
108 gr->device_set.insert(device); in CompleteGroupLocal()
120 << " dev_set=" << gr->device_set.size(); in CompleteGroupLocal()
128 << gr->device_set.size() << " gr " << gr; in CompleteGroupLocal()
130 if (gr->device_set.size() < gr->group.group_size) { in CompleteGroupLocal()
134 CHECK_EQ(gr->device_set.size(), gr->group.group_size); in CompleteGroupLocal()
Dgraph_execution_state.cc62 device_set_(options.device_set), in GraphExecutionState()
202 combined_options.device_set = device_set_; in Extend()
292 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice() argument
297 *out_device_attrs = &device_set.client_device()->attributes(); in LookupDevice()
302 *out_device_attrs = &device_set.client_device()->attributes(); in LookupDevice()
311 Device* device = device_set.FindDeviceByName( in LookupDevice()
569 optimization_options.device_set = device_set_; in InitBaseGraph()
785 optimization_options.device_set = device_set_; in BuildGraph()
Dprocess_function_library_runtime.cc305 const std::vector<string>& output_devices, const DeviceSet& device_set, in PinArgsAndRets() argument
362 device_set.FindMatchingDevices(parsed, &matching_devices); in PinArgsAndRets()
568 DeviceSet device_set; in InstantiateMultiDevice() local
570 device_set.AddDevice(d); in InstantiateMultiDevice()
574 options.input_devices, options.output_devices, device_set, graph.get())); in InstantiateMultiDevice()
589 optimization_options.device_set = &device_set; in InstantiateMultiDevice()
613 Placer placer(graph.get(), &device_set, default_device, in InstantiateMultiDevice()
629 &data->overlay_lib_, device_set, cpu_device, &graph); in InstantiateMultiDevice()
651 PartitionFunctionGraph(device_set, std::move(graph), &subgraphs)); in InstantiateMultiDevice()
Dpartitioning_utils.h34 const DeviceSet& device_set, std::unique_ptr<Graph> graph,
Doptimization_registry.h49 const DeviceSet* device_set = nullptr; // Not owned. member
Dgraph_execution_state.h42 const DeviceSet* device_set = nullptr; member
Dcolocation_graph.h170 ColocationGraph(const Graph* graph, const DeviceSet* device_set,
Dcolocation_graph.cc453 const DeviceSet* device_set, in ColocationGraph() argument
458 device_set_(device_set), in ColocationGraph()
459 device_types_(device_set->PrioritizedDeviceTypeList()), in ColocationGraph()
Dcollective_param_resolver_local.h70 std::set<string> device_set GUARDED_BY(mu);
Dprocess_function_library_runtime.h244 const DeviceSet& device_set, Graph* graph) const;
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster.cc46 VirtualCluster::VirtualCluster(const DeviceSet* device_set) in VirtualCluster() argument
48 device_set_ = device_set; in VirtualCluster()
Dvirtual_cluster.h40 VirtualCluster(const DeviceSet* device_set);
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient_library.cc83 se::Platform* platform, const absl::optional<std::set<int>>& device_set) { in GetOrCreateLocalClient() argument
86 default_options.set_allowed_devices(device_set); in GetOrCreateLocalClient()
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass_test.cc68 std::unique_ptr<DeviceSet> device_set(new DeviceSet()); in IncreaseDynamismForAutoJit() local
70 device_set->AddDevice(device.get()); in IncreaseDynamismForAutoJit()
80 options.device_set = device_set.get(); in IncreaseDynamismForAutoJit()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.h133 FunctionLibraryDefinition* lib, const DeviceSet& device_set,
Dmeta_optimizer.cc716 FunctionLibraryDefinition* flib, const DeviceSet& device_set, in OptimizeGraph() argument
730 for (const Device* d : device_set.devices()) { in OptimizeGraph()
749 tensorflow::grappler::VirtualCluster cluster(&device_set); in OptimizeGraph()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc261 std::unique_ptr<DeviceSet> device_set, in Init()
265 std::move(worker_cache), std::move(device_set), in Init()

12