| /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 arg_index) in ArgFeedRewrite()
144 const DeviceAttributes* device_info, int32 retval_index) in RetvalFetchRewrite()
|
| D | subgraph_test.cc | 141 DeviceAttributes device_info_;
347 DeviceAttributes device_info; in BM_SubgraphHelper()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.h | 62 Device(Env* env, const DeviceAttributes& device_attributes);
80 const DeviceAttributes& attributes() const override { in attributes()
185 static DeviceAttributes BuildDeviceAttributes(
189 static DeviceAttributes BuildDeviceAttributes( in BuildDeviceAttributes()
211 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 | device_mgr.h | 34 class DeviceAttributes; variable
46 void ListDeviceAttributes(std::vector<DeviceAttributes>* devices) const;
|
| D | renamed_device.cc | 36 DeviceAttributes attributes(underlying->attributes()); in NewRenamedDevice()
44 const DeviceAttributes& attributes, in RenamedDevice()
|
| D | device_set_test.cc | 30 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in Dev()
34 DeviceAttributes attr; in Dev()
|
| D | local_device.h | 37 const DeviceAttributes& attributes);
|
| D | graph_execution_state.cc | 294 const tensorflow::DeviceAttributes** out_device_attrs) { in LookupDevice()
326 const DeviceAttributes* device;
454 const DeviceAttributes* device_info; in PruneGraph()
474 const DeviceAttributes* device_info; in PruneGraph()
492 const DeviceAttributes* device_info = in PruneGraph()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 45 RemoteDevice(Env* env, const DeviceAttributes& da) in RemoteDevice()
91 for (const DeviceAttributes& da : call->resp.device_attributes()) { in NewRemoteDevices()
104 DeviceAttributes da_rewritten = da; in NewRemoteDevices()
|
| D | device_resolver_distributed_test.cc | 40 gtl::FlatMap<string, DeviceAttributes>& attr_table() { return attr_table_; } in attr_table()
49 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice()
53 DeviceAttributes attr; in NewDevice()
71 std::vector<DeviceAttributes> dev_attr; in GetStatusAsync()
|
| D | collective_rma_distributed_test.cc | 48 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice()
52 DeviceAttributes attr; in NewDevice()
77 std::vector<DeviceAttributes> dev_attr; in GetStatusAsync()
|
| D | collective_param_resolver_distributed_test.cc | 36 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice()
40 DeviceAttributes attr; in NewDevice()
56 std::vector<DeviceAttributes> dev_attr; in GetStatusAsync()
|
| D | worker_cache_partial.h | 52 typedef std::unordered_map<string, DeviceAttributes> StatusMap;
|
| D | session_mgr_test.cc | 28 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice()
37 DeviceAttributes device_attributes; in MakeCPU()
|
| D | tensor_coding_test.cc | 39 const DeviceAttributes& attributes() const override { return attr_; } in attributes()
46 DeviceAttributes attr_;
|
| D | rpcbench_test.cc | 46 std::vector<DeviceAttributes>* devices) { in MakeGRPCCluster()
105 std::vector<DeviceAttributes> devices; // One per process
|
| D | device_resolver_distributed.h | 63 gtl::FlatMap<string, DeviceAttributes> attr_table_ GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_testlib.h | 59 const std::vector<DeviceAttributes>& devices() const { return devices_; } in devices()
66 std::vector<DeviceAttributes> devices_;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 32 std::vector<DeviceAttributes> devices; in GetTestDevice()
35 bool found_cpu = absl::c_any_of(devices, [&](const DeviceAttributes& device) { in GetTestDevice()
39 bool found_gpu = absl::c_any_of(devices, [&](const DeviceAttributes& device) { in GetTestDevice()
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.cc | 52 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice()
66 DeviceAttributes device_attributes; in Make()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 45 class DeviceAttributes; variable
233 virtual const DeviceAttributes& attributes() const;
|
| D | device_base.cc | 28 const DeviceAttributes& DeviceBase::attributes() const { in attributes()
|
| D | device_attributes.proto | 32 message DeviceAttributes { message
|
| /external/tensorflow/tensorflow/python/client/ |
| D | device_lib.py | 35 m = device_attributes_pb2.DeviceAttributes()
|