| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils_test.cc | 53 device_set_.AddDevice(d); in SetUp()
71 Placer placer(graph, &device_set_, device0_); in SwapGraph()
87 Placer placer(graph, &device_set_, device0_); in TwoDeviceSwapGraph()
101 Placer placer(subgraph, &device_set_, device0_); in SubGraph()
108 DeviceSet device_set_; member in tensorflow::__anone4792f6c0111::PartitioningUtilsTest
117 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
128 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
143 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
|
| D | graph_execution_state.cc | 62 device_set_(options.device_set), in GraphExecutionState()
202 combined_options.device_set = device_set_; in Extend()
455 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, feed, in PruneGraph()
475 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, fetch, in PruneGraph()
493 &device_set_->client_device()->attributes(); in PruneGraph()
569 optimization_options.device_set = device_set_; in InitBaseGraph()
574 Placer placer(new_graph.get(), device_set_, /* default_device= */ nullptr, in InitBaseGraph()
684 for (const auto& device : device_set_->devices()) { in OptimizeGraph()
691 grappler::VirtualCluster cluster(device_set_); in OptimizeGraph()
785 optimization_options.device_set = device_set_; in BuildGraph()
|
| D | colocation_graph.cc | 458 device_set_(device_set), in ColocationGraph()
713 device_set_->FindMatchingDevices( in GetDevicesForNode()
733 device_set_->FindMatchingDevices(soft_device_name, &devices); in GetDevicesForNode()
756 device_set_->FindMatchingDevices(specified_device_name, in GetDevicesForNode()
762 for (const Device* device : device_set_->devices()) { in GetDevicesForNode()
810 if (device_set_->devices().empty()) { in GetDevicesForNode()
814 device_set_->devices(), members_[node_root].supported_device_types(), in GetDevicesForNode()
922 device_set_->FindDeviceByName(assigned_device_name); in InitializeMemberWithAssignedDevice()
955 for (Device* d : device_set_->devices()) { in InitializeMember()
|
| D | graph_execution_state.h | 195 const DeviceSet* device_set_; // Not owned variable
|
| D | colocation_graph.h | 244 const DeviceSet* device_set_; // Not owned. variable
|
| D | direct_session.cc | 320 device_set_.AddDevice(d); in DirectSession()
326 device_set_.set_client_device(d); in DirectSession()
370 options.device_set = &device_set_; in MaybeInitializeExecutionState()
1332 input, device_set_.client_device()->attributes(), FrameAndIter(0, 0)); in CreateExecutors()
1337 GetRendezvousKey(output, device_set_.client_device()->attributes(), in CreateExecutors()
1473 prune_options.device_set = &device_set_; in CreateGraphs()
|
| D | direct_session.h | 318 DeviceSet device_set_; variable
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.h | 51 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
55 const DeviceSet* device_set_ = nullptr; variable
|
| D | single_machine.h | 46 const DeviceSet* GetDeviceSet() const override { return device_set_.get(); } in GetDeviceSet()
78 std::unique_ptr<DeviceSet> device_set_; variable
|
| D | virtual_cluster.cc | 48 device_set_ = device_set; in VirtualCluster()
49 for (const auto& device : device_set_->devices()) { in VirtualCluster()
|
| D | single_machine.cc | 372 device_set_.reset(new DeviceSet); in ResetSession()
376 device_set_->AddDevice(d); in ResetSession()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 131 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
133 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
147 const DeviceSet* device_set_; member in tensorflow::tensorrt::convert::FakeCluster
|