| /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()
44 void DeviceSet::FindMatchingDevices(const DeviceNameUtils::ParsedName& spec, in FindMatchingDevices()
56 Device* DeviceSet::FindDeviceByName(const string& name) const { in FindDeviceByName()
61 int DeviceSet::DeviceTypeOrder(const DeviceType& d) { in DeviceTypeOrder()
68 auto a_priority = DeviceSet::DeviceTypeOrder(a); in DeviceTypeComparator()
69 auto b_priority = DeviceSet::DeviceTypeOrder(b); in DeviceTypeComparator()
77 std::vector<DeviceType> DeviceSet::PrioritizedDeviceTypeList() const { in PrioritizedDeviceTypeList()
90 void DeviceSet::SortPrioritizedDeviceTypeVector( in SortPrioritizedDeviceTypeVector()
[all …]
|
| D | placer.h | 72 const FunctionLibraryDefinition* flib_def, const DeviceSet* devices,
76 Placer(Graph* graph, const string& function_name, const DeviceSet* devices,
79 Placer(Graph* graph, const string& function_name, const DeviceSet* devices);
99 const DeviceSet* const devices_; // Not owned.
|
| D | device_set.h | 34 class DeviceSet {
36 DeviceSet();
37 ~DeviceSet();
129 TF_DISALLOW_COPY_AND_ASSIGN(DeviceSet);
|
| D | device_set_test.cc | 50 const DeviceSet& device_set() const { return devices_; } in device_set()
57 DeviceSet devices_;
78 EXPECT_EQ(50, DeviceSet::DeviceTypeOrder(DeviceType("d1"))); in TEST_F()
79 EXPECT_EQ(51, DeviceSet::DeviceTypeOrder(DeviceType("d2"))); in TEST_F()
80 EXPECT_EQ(49, DeviceSet::DeviceTypeOrder(DeviceType("d3"))); in TEST_F()
|
| D | inspecting_placer.h | 74 const DeviceSet* device_set, const Device* default_device,
85 const DeviceSet& device_set_;
|
| D | function_optimization_registry_pass_failure_test.cc | 31 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
47 DeviceSet device_set; in TEST()
|
| D | function_optimization_registration_test.cc | 30 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
47 DeviceSet device_set; in TEST()
|
| D | function_optimization_registry_test.cc | 32 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 DeviceSet device_set; in TEST()
|
| D | function_optimization_registry.h | 44 virtual Status Run(const DeviceSet& device_set,
63 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto,
|
| D | placer.cc | 136 const DeviceSet* devices, const Device* default_local_device, in Placer()
147 const DeviceSet* devices, const Device* default_local_device) in Placer()
152 const DeviceSet* devices) in Placer()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util.h | 33 class DeviceSet; variable
56 friend class DeviceSet; variable
60 class DeviceSet {
63 void UnionWith(const DeviceSet& other);
147 string DebugString(const DeviceSet& device_set) const;
203 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu);
212 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu);
|
| D | device_util.cc | 26 void DeviceSet::Insert(DeviceId device_id) { in Insert()
37 void DeviceSet::UnionWith(const DeviceSet& other) { in UnionWith()
47 bool DeviceSet::IsEmpty() const { in IsEmpty()
80 string DeviceInfoCache::DebugString(const DeviceSet& device_set) const { in DebugString()
102 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu, in PickDeviceForXlaImpl()
220 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { in PickDeviceForXla()
230 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { in MaybePickDeviceForXla()
|
| /external/llvm-project/lldb/source/Plugins/Platform/MacOSX/objcxx/ |
| D | PlatformiOSSimulatorCoreSimulatorSupport.h | 176 friend class DeviceSet; variable
195 class DeviceSet {
197 static DeviceSet GetAllDevices(const char *developer_dir);
199 static DeviceSet GetAvailableDevices(const char *developer_dir);
207 DeviceSet GetDevicesIf(std::function<bool(Device)> f);
209 DeviceSet GetDevices(DeviceType::ProductFamilyID dev_id);
214 DeviceSet(id arr) : m_dev(arr) {} in DeviceSet() function
|
| D | PlatformiOSSimulatorCoreSimulatorSupport.mm | 548 CoreSimulatorSupport::DeviceSet
549 CoreSimulatorSupport::DeviceSet::GetAllDevices(const char *developer_dir) {
551 return DeviceSet([NSArray new]);
561 return DeviceSet([NSArray new]);
563 return DeviceSet([[serviceContext defaultDeviceSetWithError:&error] devices]);
566 CoreSimulatorSupport::DeviceSet
567 CoreSimulatorSupport::DeviceSet::GetAvailableDevices(
575 size_t CoreSimulatorSupport::DeviceSet::GetNumDevices() {
580 CoreSimulatorSupport::DeviceSet::GetDeviceAtIndex(size_t idx) {
586 CoreSimulatorSupport::DeviceSet CoreSimulatorSupport::DeviceSet::GetDevicesIf(
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.h | 40 const DeviceSet& device_set,
51 const DeviceSet& device_set, std::vector<Device*>* host_system_devices);
62 const DeviceSet& device_set,
86 const string& node_type, Graph* graph, const DeviceSet& device_set,
|
| D | distributed_tpu_rewrite_helpers.cc | 30 const string& system_spec_string, const DeviceSet& device_set, in GetSystemDevice()
97 const DeviceNameUtils::ParsedName& system_spec, const DeviceSet& device_set, in GetHostSystemDevices()
151 const DeviceNameUtils::ParsedName& system_spec, const DeviceSet& device_set, in GetTPUDevices()
201 const string& node_type, Graph* graph, const DeviceSet& device_set, in ForConfigurationNodeMatchingType()
|
| /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,
|
| /external/angle/src/libANGLE/ |
| D | Device.cpp | 40 typedef std::set<egl::Device *> DeviceSet; typedef
41 static DeviceSet *GetDeviceSet() in GetDeviceSet()
43 static angle::base::NoDestructor<DeviceSet> devices; in GetDeviceSet()
76 const DeviceSet *deviceSet = GetDeviceSet(); in IsValidDevice()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.h | 41 explicit VirtualCluster(const DeviceSet* device_set);
53 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
57 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_;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_bridge_pass.h | 33 MlirOptimizationPassState GetPassState(const DeviceSet* device_set,
50 bool IsEnabled(const DeviceSet* device_set, const ConfigProto& config_proto,
|
| D | mlir_bridge_pass.cc | 67 bool HasTPUDevice(const DeviceSet& device_set) { in HasTPUDevice()
87 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
141 bool MlirBridgeV1CompatPass::IsEnabled(const DeviceSet* device_set, in IsEnabled()
|
| /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;
118 DeviceSet device_set; in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.h | 67 const DeviceSet* device_set, const ConfigProto& config_proto,
134 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto,
160 virtual bool IsEnabled(const DeviceSet* device_set,
|