| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass_test.cc | 130 DeviceSet device_set_; member in tensorflow::MlirGraphOptimizationPassTest
145 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
160 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
174 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
190 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
209 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
226 device_set_, config_proto_, &graph_, flib_.get(), in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils_test.cc | 54 device_set_.AddDevice(d); in SetUp()
72 Placer placer(graph, "", &device_set_, device0_); in SwapGraph()
88 Placer placer(graph, "", &device_set_, device0_); in TwoDeviceSwapGraph()
108 Placer placer(subgraph, "", &device_set_, device0_); in SubGraph()
115 DeviceSet device_set_; member in tensorflow::__anone51cedca0111::PartitioningUtilsTest
124 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
135 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
150 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
|
| D | graph_execution_state.cc | 78 device_set_(options.device_set), in GraphExecutionState()
260 combined_options.device_set = device_set_; in Extend()
514 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, feed, in PruneGraph()
534 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, fetch, in PruneGraph()
552 &device_set_->client_device()->attributes(); in PruneGraph()
615 optimization_options.device_set = device_set_; in InitBaseGraph()
620 Placer placer(new_graph.get(), "", flib_def_.get(), device_set_, in InitBaseGraph()
659 for (const Device* d : device_set_->devices()) { in OptimizeGraph()
793 grappler::VirtualCluster cluster(device_set_); in OptimizeGraph()
795 for (const auto& device : device_set_->devices()) { in OptimizeGraph()
[all …]
|
| D | colocation_graph.cc | 616 device_set_(*device_set), in ColocationGraph()
1115 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates()
1138 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates()
1189 device_set_.FindMatchingDevices(root_member.requested_device_name(), in GetDevicesForNode()
1218 device_set_.FindMatchingDevices(specified_device_name, in GetDevicesForNode()
1224 for (const Device* device : device_set_.devices()) { in GetDevicesForNode()
1271 absl::StrJoin(DevicesToString(device_set_.devices()), ", "), "]. ", in GetDevicesForNode()
1278 if (device_set_.devices().empty()) { in GetDevicesForNode()
1281 devices = FilterSupportedDevices(device_set_.devices(), in GetDevicesForNode()
1400 device_set_.FindDeviceByName(assigned_device_name); in InitializeMemberWithAssignedDevice()
[all …]
|
| D | inspecting_placer.cc | 121 device_set_(*device_set), in InspectingPlacer()
149 fbody->graph, stack_.Push(&node, func.name()), &flib_def_, &device_set_, in ComputeIOColocationGroups()
|
| D | inspecting_placer.h | 85 const DeviceSet& device_set_; variable
|
| D | process_function_library_runtime.h | 207 return device_set_; in device_set()
223 device_set_->AddDevice(d); in AddCompositeDevice()
457 std::shared_ptr<DeviceSet> device_set_ TF_GUARDED_BY(mu_);
|
| D | graph_execution_state.h | 198 const DeviceSet* device_set_; // Not owned variable
|
| D | colocation_graph.h | 374 const DeviceSet& device_set_; variable
|
| D | direct_session.cc | 366 device_set_.AddDevice(d); in DirectSession()
372 device_set_.set_client_device(d); in DirectSession()
439 options.device_set = &device_set_; in ExtendLocked()
1449 input, device_set_.client_device()->attributes(), FrameAndIter(0, 0)); in CreateExecutors()
1454 GetRendezvousKey(output, device_set_.client_device()->attributes(), in CreateExecutors()
1598 prune_options.device_set = &device_set_; in CreateGraphs()
|
| D | direct_session.h | 341 DeviceSet device_set_; variable
|
| D | process_function_library_runtime.cc | 219 device_set_ = std::make_shared<DeviceSet>(); in InitializeDeviceAndFlr()
221 device_set_->AddDevice(d); in InitializeDeviceAndFlr()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.h | 53 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
57 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 | 369 device_set_.reset(new DeviceSet); in ResetSession()
373 device_set_->AddDevice(d); in ResetSession()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 58 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
60 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
74 const DeviceSet* device_set_ = nullptr; member in tensorflow::tensorrt::convert::FakeCluster
|