| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_set.cc | 29 DeviceSet::DeviceSet() {} in DeviceSet() function in tensorflow::DeviceSet
31 DeviceSet::~DeviceSet() {} in ~DeviceSet()
33 void DeviceSet::AddDevice(Device* device) { in AddDevice()
41 void DeviceSet::FindMatchingDevices(const DeviceNameUtils::ParsedName& spec, in FindMatchingDevices()
53 Device* DeviceSet::FindDeviceByName(const string& name) const { in FindDeviceByName()
58 int DeviceSet::DeviceTypeOrder(const DeviceType& d) { in DeviceTypeOrder()
65 auto a_priority = DeviceSet::DeviceTypeOrder(a); in DeviceTypeComparator()
66 auto b_priority = DeviceSet::DeviceTypeOrder(b); in DeviceTypeComparator()
74 std::vector<DeviceType> DeviceSet::PrioritizedDeviceTypeList() const { in PrioritizedDeviceTypeList()
|
| D | placer.h | 71 Placer(Graph* graph, const DeviceSet* devices, const Device* default_device,
74 Placer(Graph* graph, const DeviceSet* devices, const Device* default_device);
76 Placer(Graph* graph, const DeviceSet* devices);
94 const DeviceSet* const devices_; // Not owned.
|
| D | device_set.h | 32 class DeviceSet {
34 DeviceSet();
35 ~DeviceSet();
84 TF_DISALLOW_COPY_AND_ASSIGN(DeviceSet);
|
| D | device_set_test.cc | 53 DeviceSet devices_;
71 EXPECT_EQ(50, DeviceSet::DeviceTypeOrder(DeviceType("d1"))); in TEST_F()
72 EXPECT_EQ(51, DeviceSet::DeviceTypeOrder(DeviceType("d2"))); in TEST_F()
73 EXPECT_EQ(49, DeviceSet::DeviceTypeOrder(DeviceType("d3"))); in TEST_F()
|
| D | placer.cc | 81 Placer::Placer(Graph* graph, const DeviceSet* devices, in Placer()
90 Placer::Placer(Graph* graph, const DeviceSet* devices, in Placer()
94 Placer::Placer(Graph* graph, const DeviceSet* devices) in Placer()
|
| D | graph_execution_state.h | 42 const DeviceSet* device_set = nullptr;
195 const DeviceSet* device_set_; // Not owned
|
| D | colocation_graph.h | 170 ColocationGraph(const Graph* graph, const DeviceSet* device_set,
244 const DeviceSet* device_set_; // Not owned.
|
| D | colocation_graph.cc | 78 DeviceSet::DeviceTypeOrder(DeviceType(a.first->device_type())); in FilterSupportedDevices()
80 DeviceSet::DeviceTypeOrder(DeviceType(b.first->device_type())); in FilterSupportedDevices()
357 auto a_priority = DeviceSet::DeviceTypeOrder(a.first); in MergeSupportedDevices()
358 auto b_priority = DeviceSet::DeviceTypeOrder(b.first); in MergeSupportedDevices()
453 const DeviceSet* device_set, in ColocationGraph()
|
| D | partitioning_utils.h | 34 const DeviceSet& device_set, std::unique_ptr<Graph> graph,
|
| D | optimization_registry.h | 49 const DeviceSet* device_set = nullptr; // Not owned.
|
| D | partitioning_utils.cc | 26 const DeviceSet& device_set, std::unique_ptr<Graph> graph, in PartitionFunctionGraph()
|
| D | placer_test.cc | 242 Status Place(Graph* graph, DeviceSet* devices, bool allow_soft_placement, in Place()
249 Status Place(Graph* graph, DeviceSet* devices) { in Place()
271 DeviceSet devices_;
1241 DeviceSet empty; in TEST_F()
1261 DeviceSet heterogeneous; in TEST_F()
1362 DeviceSet cpu_only; in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.h | 71 GreedyScheduler(const DeviceSet* devices, const CostModel* cost_model,
84 const DeviceSet* devices_;
95 PriorityScheduler(const DeviceSet* devices, const CostModel* cost_model,
109 const DeviceSet* devices_;
|
| D | master_env.h | 31 class DeviceSet; variable
87 std::unique_ptr<DeviceSet> device_set,
|
| D | master_session.h | 54 std::unique_ptr<DeviceSet> device_set,
132 std::unique_ptr<DeviceSet> devices_;
|
| D | scheduler.cc | 189 GreedyScheduler::GreedyScheduler(const DeviceSet* devices, in GreedyScheduler()
293 PriorityScheduler::PriorityScheduler(const DeviceSet* devices, in PriorityScheduler()
|
| D | master.cc | 372 std::unique_ptr<DeviceSet> device_set; in CreateSession()
429 device_set.reset(new DeviceSet); in CreateSession()
447 device_set.reset(new DeviceSet); in CreateSession()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.h | 40 VirtualCluster(const DeviceSet* device_set);
51 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
55 const DeviceSet* device_set_ = nullptr;
|
| D | single_machine.h | 46 const DeviceSet* GetDeviceSet() const override { return device_set_.get(); } in GetDeviceSet()
78 std::unique_ptr<DeviceSet> device_set_;
|
| D | cluster.h | 104 virtual const DeviceSet* GetDeviceSet() const { return nullptr; } in GetDeviceSet()
|
| D | virtual_cluster.cc | 46 VirtualCluster::VirtualCluster(const DeviceSet* device_set) in VirtualCluster()
|
| /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_;
191 DeviceSet device_set; in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.h | 133 FunctionLibraryDefinition* lib, const DeviceSet& device_set,
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.cc | 125 DeviceSet device_set; in RunMain()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 40 class DeviceSet; variable
577 FunctionLibraryDefinition*, const DeviceSet&,
|