| /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.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 | remote_device.h | 27 class DeviceAttributes; variable
42 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
65 DeviceAttributes device_attribute);
|
| D | collective_rma_distributed.cc | 46 const DeviceAttributes& server_attributes, in RecvBufCall()
125 DeviceAttributes server_attributes; in RecvFromPeer()
131 DeviceAttributes server_attributes; in RecvFromPeer()
262 std::vector<DeviceAttributes> cached_attrs; in CheckPeerHealth()
268 for (const DeviceAttributes& da : resp->device_attributes()) { in CheckPeerHealth()
271 for (const DeviceAttributes& attr : cached_attrs) { in CheckPeerHealth()
|
| D | session_mgr_test.cc | 31 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice()
40 DeviceAttributes device_attributes; in MakeCPU()
90 protobuf::RepeatedPtrField<DeviceAttributes> cluster_device_attributes; in TEST_F()
91 DeviceAttributes* local_cpu = cluster_device_attributes.Add(); in TEST_F()
93 DeviceAttributes* remote_cpu = cluster_device_attributes.Add(); in TEST_F()
166 protobuf::RepeatedPtrField<DeviceAttributes> cluster_device_attributes; in TEST_F()
219 protobuf::RepeatedPtrField<DeviceAttributes> cluster_device_attributes; in TEST_F()
|
| D | collective_param_resolver_distributed.h | 37 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
41 void CompleteGroupAsync(const DeviceAttributes& device,
67 void CompleteGroupDistributed(const DeviceAttributes& device,
|
| D | session_mgr.h | 61 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
73 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
85 const protobuf::RepeatedPtrField<DeviceAttributes>&
|
| D | collective_param_resolver_distributed.cc | 35 const DeviceAttributes& device, in CompleteGroupCall()
108 const DeviceAttributes& device, CollectiveParams* cp, in CompleteParamsAsync()
117 std::vector<DeviceAttributes> devices; in CompleteParamsAsync()
143 const DeviceAttributes& device, CollGroupParams* group_params, in CompleteGroupAsync()
235 for (const DeviceAttributes& device : resp.device_attributes()) { in UpdateGroupCache()
274 const DeviceAttributes& device, CollGroupParams* group_params, in CompleteGroupDistributed()
|
| D | session_mgr.cc | 74 const protobuf::RepeatedPtrField<DeviceAttributes>& in CreateSession()
85 const protobuf::RepeatedPtrField<DeviceAttributes>& in CreateSession()
232 const protobuf::RepeatedPtrField<DeviceAttributes>& in UpdateSession()
262 std::vector<DeviceAttributes> added_cluster_device_attrs; in UpdateSession()
282 protobuf::RepeatedPtrField<DeviceAttributes> added_cluster_device_attrs_pb( in UpdateSession()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.h | 53 PruneRewrite(const string* endpoint_name, const DeviceAttributes* device_info) in PruneRewrite()
67 const DeviceAttributes& device_info() { return *device_info_; } in device_info()
71 const DeviceAttributes* const device_info_; // Not owned.
104 const DeviceAttributes& device_info, bool use_function_convention,
123 const DeviceAttributes* device_info, int32_t arg_index) in ArgFeedRewrite()
144 const DeviceAttributes* device_info, int32_t retval_index) in RetvalFetchRewrite()
|
| /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()
198 DeviceAttributes device_attributes_;
|
| D | collective.h | 63 DeviceAttributes device;
169 DeviceAttributes* attributes) = 0;
173 const string& task, std::vector<DeviceAttributes>* attributes) = 0;
178 const std::vector<DeviceAttributes>& attributes) = 0;
189 virtual void CompleteParamsAsync(const DeviceAttributes& device,
196 virtual void CompleteGroupAsync(const DeviceAttributes& device,
323 virtual void CompleteParamsAsync(const DeviceAttributes& device, in CompleteParamsAsync()
332 virtual void CompleteGroupAsync(const DeviceAttributes& device, in CompleteGroupAsync()
|
| D | device.cc | 27 Device::Device(Env* env, const DeviceAttributes& device_attributes) in Device()
43 DeviceAttributes Device::BuildDeviceAttributes( in BuildDeviceAttributes()
46 DeviceAttributes da; in BuildDeviceAttributes()
|
| /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 | device_mgr.h | 35 class DeviceAttributes; variable
45 std::vector<DeviceAttributes>* devices) const = 0;
90 std::vector<DeviceAttributes>* devices) const override;
130 std::vector<DeviceAttributes>* devices) const override;
|
| D | collective_param_resolver_local.h | 51 void CompleteParamsAsync(const DeviceAttributes& device, CollectiveParams* cp,
55 void CompleteGroupAsync(const DeviceAttributes& device,
90 void CompleteGroupLocal(const DeviceAttributes& device,
|
| D | device_mgr_test.cc | 33 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice()
37 DeviceAttributes attr; in CreateDevice()
|
| D | renamed_device.cc | 42 DeviceAttributes attributes(underlying->attributes()); in NewRenamedDevice()
51 const DeviceAttributes& attributes, in RenamedDevice()
|
| /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()
|