| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 40 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice()
44 DeviceAttributes attr; in NewDevice()
63 std::vector<DeviceAttributes> attributes; in SetUp()
78 DeviceAttributes attributes; in TEST_F()
85 DeviceAttributes attributes; in TEST_F()
91 std::vector<DeviceAttributes> attributes; in TEST_F()
96 Property(&DeviceAttributes::name, in TEST_F()
98 Property(&DeviceAttributes::name, in TEST_F()
104 Property(&DeviceAttributes::name, in TEST_F()
106 Property(&DeviceAttributes::name, in TEST_F()
[all …]
|
| D | device_resolver_distributed.h | 35 DeviceAttributes* attributes) override;
38 const string& task, std::vector<DeviceAttributes>* attributes) override;
41 const std::vector<DeviceAttributes>& attributes) override;
46 absl::flat_hash_map<string, DeviceAttributes> attr_table_ TF_GUARDED_BY(mu_);
|
| D | device_resolver_distributed.cc | 31 const string& device, DeviceAttributes* attributes) { in GetDeviceAttributes()
42 const string& task, std::vector<DeviceAttributes>* attributes) { in GetAllDeviceAttributes()
58 const std::vector<DeviceAttributes>& attributes) { in UpdateDeviceAttributes()
60 for (const DeviceAttributes& attr : attributes) { in UpdateDeviceAttributes()
|
| D | remote_device.h | 27 class DeviceAttributes; variable
42 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
65 DeviceAttributes device_attribute);
|
| D | remote_device.cc | 38 RemoteDevice(Env* env, const DeviceAttributes& da) in RemoteDevice()
63 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes, in AsRemoteDevices()
113 for (const DeviceAttributes& da : call->resp.device_attributes()) { in NewRemoteDevices()
126 DeviceAttributes da_rewritten = da; in NewRemoteDevices()
153 DeviceAttributes device_attribute) { in NewRemoteDevice()
|
| D | session_mgr_test.cc | 28 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice()
37 DeviceAttributes device_attributes; in MakeCPU()
87 protobuf::RepeatedPtrField<DeviceAttributes> cluster_device_attributes; in TEST_F()
88 DeviceAttributes* local_cpu = cluster_device_attributes.Add(); in TEST_F()
90 DeviceAttributes* remote_cpu = cluster_device_attributes.Add(); in TEST_F()
163 protobuf::RepeatedPtrField<DeviceAttributes> cluster_device_attributes; in TEST_F()
218 protobuf::RepeatedPtrField<DeviceAttributes> cluster_device_attributes; in TEST_F()
|
| D | session_mgr.h | 54 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
66 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
74 const protobuf::RepeatedPtrField<DeviceAttributes>&
|
| D | collective_rma_distributed.cc | 42 const DeviceAttributes& server_attributes, in RecvBufCall()
95 DeviceAttributes server_attributes; in RecvFromPeer()
218 std::vector<DeviceAttributes> cached_attrs; in CheckPeerHealth()
224 for (const DeviceAttributes& da : resp->device_attributes()) { in CheckPeerHealth()
227 for (const DeviceAttributes& attr : cached_attrs) { in CheckPeerHealth()
|
| D | collective_param_resolver_distributed.h | 37 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
67 void CompleteGroupDistributed(const DeviceAttributes& device,
|
| D | session_mgr.cc | 62 const protobuf::RepeatedPtrField<DeviceAttributes>& in CreateSession()
72 const protobuf::RepeatedPtrField<DeviceAttributes>& in CreateSession()
192 const protobuf::RepeatedPtrField<DeviceAttributes>& in UpdateSession()
223 std::vector<DeviceAttributes> added_cluster_device_attrs; in UpdateSession()
243 protobuf::RepeatedPtrField<DeviceAttributes> added_cluster_device_attrs_pb( in UpdateSession()
|
| D | collective_param_resolver_distributed.cc | 35 const DeviceAttributes& device, in CompleteGroupCall()
108 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
116 std::vector<DeviceAttributes> attributes; in CompleteParamsAsync()
251 for (const DeviceAttributes& device : resp.device_attributes()) { in UpdateGroupCache()
287 const DeviceAttributes& device, CollectiveParams* cp, in CompleteGroupDistributed()
|
| D | collective_rma_distributed_test.cc | 49 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice()
53 DeviceAttributes attr; in NewDevice()
84 std::vector<DeviceAttributes> dev_attr; in GetStatusAsync()
250 std::vector<DeviceAttributes>* dv = &dev_by_task_[worker_name]; in DefineWorker()
287 std::unordered_map<string, std::vector<DeviceAttributes>> dev_by_task_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device.h | 59 Device(Env* env, const DeviceAttributes& device_attributes);
79 const DeviceAttributes& attributes() const override { in attributes()
165 static DeviceAttributes BuildDeviceAttributes(
169 static DeviceAttributes BuildDeviceAttributes( in BuildDeviceAttributes()
191 const DeviceAttributes device_attributes_;
|
| D | device.cc | 26 Device::Device(Env* env, const DeviceAttributes& device_attributes) in Device()
42 DeviceAttributes Device::BuildDeviceAttributes( in BuildDeviceAttributes()
45 DeviceAttributes da; in BuildDeviceAttributes()
|
| D | collective.h | 160 DeviceAttributes* attributes) = 0;
164 const string& task, std::vector<DeviceAttributes>* attributes) = 0;
169 const std::vector<DeviceAttributes>& attributes) = 0;
180 virtual void CompleteParamsAsync(const DeviceAttributes& device,
308 virtual void CompleteParamsAsync(const DeviceAttributes& device, in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_resolver_local_test.cc | 48 DeviceAttributes attributes; in TEST_F()
55 DeviceAttributes attributes; in TEST_F()
61 std::vector<DeviceAttributes> attributes; in TEST_F()
67 std::vector<DeviceAttributes> attributes; in TEST_F()
|
| D | device_resolver_local.h | 34 DeviceAttributes* attributes) override;
37 const string& task, std::vector<DeviceAttributes>* attributes) override;
40 const std::vector<DeviceAttributes>& attributes) override;
|
| D | device_resolver_local.cc | 23 DeviceAttributes* attributes) { in GetDeviceAttributes()
37 const string& task, std::vector<DeviceAttributes>* attributes) { in GetAllDeviceAttributes()
43 const std::vector<DeviceAttributes>& attributes) { in UpdateDeviceAttributes()
|
| D | collective_param_resolver_local.h | 50 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
75 std::unordered_map<string, DeviceAttributes> devices TF_GUARDED_BY(mu);
87 void CompleteGroupLocal(const DeviceAttributes& device, CollectiveParams* cp,
144 void CompleteDefaultRanking(const std::vector<DeviceAttributes>& attributes,
|
| D | device_mgr.h | 35 class DeviceAttributes; variable
45 std::vector<DeviceAttributes>* devices) const = 0;
90 std::vector<DeviceAttributes>* devices) const override;
126 std::vector<DeviceAttributes>* devices) const override;
|
| D | renamed_device.cc | 42 DeviceAttributes attributes(underlying->attributes()); in NewRenamedDevice()
51 const DeviceAttributes& attributes, in RenamedDevice()
|
| D | device_mgr_test.cc | 33 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice()
37 DeviceAttributes attr; in CreateDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_testlib.h | 64 const std::vector<DeviceAttributes>& devices() const { return devices_; } in devices()
71 std::vector<DeviceAttributes> devices_;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 46 std::vector<DeviceAttributes> devices; in GetTestDevice()
49 bool found_cpu = absl::c_any_of(devices, [&](const DeviceAttributes& device) { in GetTestDevice()
53 bool found_gpu = absl::c_any_of(devices, [&](const DeviceAttributes& device) { in GetTestDevice()
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.cc | 49 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice()
63 DeviceAttributes device_attributes; in Make()
|