| /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 const FunctionLibraryDefinition* flib_def, const DeviceSet* devices);
78 const FunctionLibraryDefinition* flib_def, 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_test.cc | 32 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 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_pass_failure_test.cc | 31 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
47 DeviceSet device_set; in TEST()
|
| D | partitioning_utils.h | 37 const DeviceSet& device_set, std::unique_ptr<Graph> graph,
44 const DeviceSet& device_set, std::unique_ptr<Graph> graph);
|
| D | function_optimization_registry.h | 44 virtual Status Run(const DeviceSet& device_set,
63 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto,
|
| /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);
145 string DebugString(const DeviceSet& device_set) const;
201 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu);
210 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu);
|
| D | device_util.cc | 25 void DeviceSet::Insert(DeviceId device_id) { in Insert()
36 void DeviceSet::UnionWith(const DeviceSet& other) { in UnionWith()
46 bool DeviceSet::IsEmpty() const { in IsEmpty()
79 string DeviceInfoCache::DebugString(const DeviceSet& device_set) const { in DebugString()
101 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu, in PickDeviceForXlaImpl()
219 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { in PickDeviceForXla()
229 const jit::DeviceSet& devices, bool allow_mixing_unknown_and_cpu) { in MaybePickDeviceForXla()
|
| /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_;
|
| /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;
|
| /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/tpu/ |
| D | tpu_global_init.cc | 71 DeviceSet* device_set) { in DeviceSetFromDeviceMgr()
91 const DeviceSet& device_set, in ConstructDistributedInitializationGraph()
139 const DeviceSet& device_set, Env* env, in InitializeTPUSystemGlobally()
205 DeviceSet device_set; in InitializeTPUSystemGlobally()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt_distributed_interface.h | 22 class DeviceSet; variable
32 virtual void PopulateRemoteDevices(tensorflow::DeviceSet* dev_set) = 0;
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 48 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
50 const DeviceSet* GetDeviceSet() const override { return device_set_; } in GetDeviceSet()
64 const DeviceSet* device_set_ = nullptr;
107 DeviceSet device_set; in TEST()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.h | 225 const DeviceSet& device_set,
231 const DeviceSet& device_set,
248 const DeviceSet& device_set, int replica_id, uint64 cache_hash,
263 Status SetDeviceOrdinal(const DeviceSet& device_set, int device_ordinal,
311 DeviceSet device_set_;
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass_test.cc | 37 const DeviceSet* device_set,
52 const DeviceSet* device_set,
67 const DeviceSet* device_set,
148 DeviceSet device_set_;
|
| D | mlir_graph_optimization_pass.h | 63 const DeviceSet* device_set, const ConfigProto& config_proto,
119 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto,
146 const DeviceSet* device_set, const ConfigProto& config_proto,
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | fallback_state.h | 54 const DeviceSet &device_set() const { return device_set_; } in device_set()
64 DeviceSet device_set_;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_bridge_pass.h | 34 const DeviceSet* device_set, const ConfigProto& config_proto,
53 const DeviceSet* device_set, const ConfigProto& config_proto,
|