Home
last modified time | relevance | path

Searched refs:SelectDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_test.cc99 Status status = context()->SelectDevice( in TEST_F()
108 status = context()->SelectDevice(requested, supported, DT_INVALID, &dev); in TEST_F()
116 TF_ASSERT_OK(context()->SelectDevice(requested, supported, DT_INVALID, &dev)); in TEST_F()
121 TF_ASSERT_OK(context()->SelectDevice(requested, supported, DT_INVALID, &dev)); in TEST_F()
126 TF_ASSERT_OK(context()->SelectDevice(requested, supported, DT_STRING, &dev)); in TEST_F()
145 Status status = context()->SelectDevice( in TEST_F()
154 TF_ASSERT_OK(context()->SelectDevice(requested, supported, DT_INVALID, &dev)); in TEST_F()
159 TF_ASSERT_OK(context()->SelectDevice(requested, supported, DT_INVALID, &dev)); in TEST_F()
164 TF_ASSERT_OK(context()->SelectDevice(requested, supported, DT_INVALID, &dev)); in TEST_F()
169 TF_ASSERT_OK(context()->SelectDevice(requested, supported, DT_STRING, &dev)); in TEST_F()
Dcontext.h176 Status SelectDevice(DeviceNameUtils::ParsedName preferred,
Dexecute.cc461 TF_RETURN_IF_ERROR(ctx.SelectDevice(op->GetDeviceParsedName(), in EagerLocalExecute()
Dcontext.cc171 Status EagerContext::SelectDevice(DeviceNameUtils::ParsedName preferred, in SelectDevice() function in tensorflow::EagerContext