| /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::__anond3194bee0111::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 | 70 device_set_(options.device_set), in GraphExecutionState()
252 combined_options.device_set = device_set_; in Extend()
506 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, feed, in PruneGraph()
526 TF_RETURN_IF_ERROR(LookupDevice(*device_set_, fetch, in PruneGraph()
544 &device_set_->client_device()->attributes(); in PruneGraph()
607 optimization_options.device_set = device_set_; in InitBaseGraph()
612 Placer placer(new_graph.get(), "", flib_def_.get(), device_set_, in InitBaseGraph()
648 for (const Device* d : device_set_->devices()) { in OptimizeGraph()
726 for (const auto& device : device_set_->devices()) { in OptimizeGraph()
733 grappler::VirtualCluster cluster(device_set_); in OptimizeGraph()
[all …]
|
| D | colocation_graph.cc | 586 device_set_(*device_set), in ColocationGraph()
986 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates()
1009 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates()
1060 device_set_.FindMatchingDevices(root_member.requested_device_name(), in GetDevicesForNode()
1089 device_set_.FindMatchingDevices(specified_device_name, in GetDevicesForNode()
1095 for (const Device* device : device_set_.devices()) { in GetDevicesForNode()
1142 absl::StrJoin(DevicesToString(device_set_.devices()), ", "), "]. ", in GetDevicesForNode()
1149 if (device_set_.devices().empty()) { in GetDevicesForNode()
1152 devices = FilterSupportedDevices(device_set_.devices(), in GetDevicesForNode()
1271 device_set_.FindDeviceByName(assigned_device_name); in InitializeMemberWithAssignedDevice()
[all …]
|
| D | inspecting_placer.cc | 120 device_set_(*device_set), in InspectingPlacer()
148 fbody->graph, stack_.Push(&node, func.name()), &flib_def_, &device_set_, in ComputeIOColocationGroups()
|
| D | inspecting_placer.h | 85 const DeviceSet& device_set_; variable
|
| D | graph_execution_state.h | 194 const DeviceSet* device_set_; // Not owned variable
|
| D | process_function_library_runtime.cc | 116 device_set_.AddDevice(d); in ProcessFunctionLibraryRuntime()
667 options.input_devices, options.output_devices, device_set_, arg_nodes, in InstantiateMultiDevice()
683 optimization_options.device_set = &device_set_; in InstantiateMultiDevice()
693 &device_set_, default_device, in InstantiateMultiDevice()
709 &data->lib_def_, device_set_, cpu_device, &graph); in InstantiateMultiDevice()
733 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs)); in InstantiateMultiDevice()
815 device_set_.FindDeviceByName(target)->device_type(); in InstantiateMultiDevice()
|
| D | process_function_library_runtime.h | 192 const DeviceSet* device_set() { return &device_set_; } in device_set()
439 DeviceSet device_set_; variable
|
| D | colocation_graph.h | 362 const DeviceSet& device_set_; variable
|
| D | direct_session.cc | 363 device_set_.AddDevice(d); in DirectSession()
369 device_set_.set_client_device(d); in DirectSession()
436 options.device_set = &device_set_; in ExtendLocked()
1436 input, device_set_.client_device()->attributes(), FrameAndIter(0, 0)); in CreateExecutors()
1441 GetRendezvousKey(output, device_set_.client_device()->attributes(), in CreateExecutors()
1587 prune_options.device_set = &device_set_; in CreateGraphs()
|
| D | direct_session.h | 341 DeviceSet device_set_; variable
|
| /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 | 57 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
59 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
73 const DeviceSet* device_set_; member in tensorflow::tensorrt::convert::FakeCluster
|