Searched refs:SelectDevice (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | placement_test.cc | 125 Status status = context()->SelectDevice(requested, invalid_op, &dev); in TEST_F()
135 status = context()->SelectDevice(requested, node, &dev); in TEST_F()
144 TF_ASSERT_OK(context()->SelectDevice(requested, node, &dev)); in TEST_F()
149 TF_ASSERT_OK(context()->SelectDevice(requested, node, &dev)); in TEST_F()
166 Status status = context()->SelectDevice(requested, invalid_op, &dev); in TEST_F()
176 status = context()->SelectDevice(requested, node, &dev); in TEST_F()
|
| D | context.h | 206 Status SelectDevice(DeviceNameUtils::ParsedName preferred,
|
| D | context.cc | 349 Status EagerContext::SelectDevice(DeviceNameUtils::ParsedName preferred, in SelectDevice() function in tensorflow::EagerContext
|
| D | execute.cc | 1030 TF_RETURN_IF_ERROR(ctx.SelectDevice(preferred_device, ndef, device)); in SetOpDevice()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.cc | 105 eager_context_->SelectDevice(device_parsed_name, *ndef, &device)); in SelectFromNodeDef()
|