Searched refs:DeviceAttributes (Results 1 – 25 of 44) sorted by relevance
12
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.h | 56 Device(Env* env, const DeviceAttributes& device_attributes);
74 const DeviceAttributes& attributes() const override { in attributes()
140 static DeviceAttributes BuildDeviceAttributes(
144 static DeviceAttributes BuildDeviceAttributes( in BuildDeviceAttributes()
161 const DeviceAttributes device_attributes_;
|
| D | device.cc | 26 Device::Device(Env* env, const DeviceAttributes& device_attributes) in Device()
40 DeviceAttributes Device::BuildDeviceAttributes( in BuildDeviceAttributes()
43 DeviceAttributes da; in BuildDeviceAttributes()
|
| D | device_mgr.h | 33 class DeviceAttributes; variable
44 void ListDeviceAttributes(std::vector<DeviceAttributes>* devices) const;
|
| D | renamed_device.cc | 37 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 | local_device.cc | 63 const DeviceAttributes& attributes) in LocalDevice()
|
| /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 | session_mgr_test.cc | 27 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice()
36 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 | worker_cache_partial.h | 52 typedef std::unordered_map<string, DeviceAttributes> StatusMap;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_testlib.h | 58 const std::vector<DeviceAttributes>& devices() const { return devices_; } in devices()
65 std::vector<DeviceAttributes> devices_;
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.h | 71 const DeviceAttributes& device_info, bool use_function_convention,
84 Status FetchOutputs(Graph* g, const DeviceAttributes& device_info,
|
| D | subgraph.cc | 56 static Status FeedInputs(Graph* g, const DeviceAttributes& device_info, in FeedInputs()
188 Status FetchOutputs(Graph* g, const DeviceAttributes& device_info, in FetchOutputs()
269 const DeviceAttributes& device_info, bool use_function_convention, in RewriteGraphForExecution()
|
| D | subgraph_test.cc | 141 DeviceAttributes device_info_;
347 DeviceAttributes device_info; in BM_SubgraphHelper()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 46 class DeviceAttributes; variable
208 virtual const DeviceAttributes& attributes() const;
|
| D | device_base.cc | 22 const DeviceAttributes& DeviceBase::attributes() const { in attributes()
|
| D | resource_mgr_test.cc | 213 const DeviceAttributes& attributes() const override { return attr_; } in attributes()
216 DeviceAttributes attr_;
|
| D | device_attributes.proto | 31 message DeviceAttributes { message
|
| /external/tensorflow/tensorflow/python/client/ |
| D | device_lib.py | 32 m = device_attributes_pb2.DeviceAttributes()
|
| D | device_lib.i | 41 const DeviceAttributes& attr = device->attributes(); in ListDevices()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 132 const DeviceAttributes attrs = Device::BuildDeviceAttributes( in Create()
179 const DeviceAttributes& attrs, int device_ordinal, in XlaDevice()
|
| D | xla_device.h | 80 XlaDevice(const SessionOptions& options, const DeviceAttributes& attrs,
|
| /external/tensorflow/tensorflow/core/public/ |
| D | session.h | 180 virtual Status ListDevices(std::vector<DeviceAttributes>* response) = 0;
|
12