Home
last modified time | relevance | path

Searched refs:ContainsDevice (Results 1 – 6 of 6) 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;
Ddevice_mgr.cc127 bool StaticDeviceMgr::ContainsDevice(int64 device_incarnation) const { in ContainsDevice() function in tensorflow::StaticDeviceMgr
Ddynamic_device_mgr.cc95 bool DynamicDeviceMgr::ContainsDevice(int64 device_incarnation) const { in ContainsDevice() function in tensorflow::DynamicDeviceMgr
Ddynamic_device_mgr_test.cc96 EXPECT_FALSE(dm->ContainsDevice(d1_incarnation)); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle_test.cc270 EXPECT_TRUE(local_device_mgr.ContainsDevice( in TEST()
276 EXPECT_FALSE(new_device_mgr.ContainsDevice( in TEST()
306 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
312 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
317 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
322 EXPECT_FALSE(remote_device_mgr.ContainsDevice( in TEST()
324 EXPECT_TRUE(remote_device_mgr.ContainsDevice( in TEST()
Dplacement_utils.cc53 if (ctx->remote_device_mgr()->ContainsDevice(device_incarnation)) { in ValidateTensorHandleRemoteDevice()