Home
last modified time | relevance | path

Searched refs:device_incarnation (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr.h63 virtual bool ContainsDevice(int64 device_incarnation) const = 0;
95 bool ContainsDevice(int64 device_incarnation) const override;
131 bool ContainsDevice(int64 device_incarnation) const override;
Dbuf_rendezvous.cc150 const uint64 device_incarnation, in ConsumeBuf() argument
158 device->attributes().incarnation() != device_incarnation) { in ConsumeBuf()
160 "RecvBuf expects a different device incarnation: ", device_incarnation, in ConsumeBuf()
Ddevice_mgr.cc127 bool StaticDeviceMgr::ContainsDevice(int64 device_incarnation) const { in ContainsDevice()
128 return device_incarnation_set_.contains(device_incarnation); in ContainsDevice()
Ddynamic_device_mgr.cc95 bool DynamicDeviceMgr::ContainsDevice(int64 device_incarnation) const { in ContainsDevice()
97 return device_incarnation_set_.contains(device_incarnation); in ContainsDevice()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dplacement_utils.cc52 int64 device_incarnation) { in ValidateTensorHandleRemoteDevice() argument
53 if (ctx->remote_device_mgr()->ContainsDevice(device_incarnation)) { in ValidateTensorHandleRemoteDevice()