Searched refs:GetHandleOnDevice (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 372 EXPECT_EQ(0, proc_flr_->GetHandleOnDevice( in TEST_F()
377 EXPECT_EQ(0, proc_flr_->GetHandleOnDevice( in TEST_F()
383 EXPECT_EQ(1, proc_flr_->GetHandleOnDevice( in TEST_F()
403 EXPECT_EQ(0, proc_flr_->GetHandleOnDevice( in TEST_F()
|
| D | function.cc | 519 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, h); in GetFunctionBody()
684 parent_->GetHandleOnDevice(device_name_, *handle); in Instantiate()
731 local_handle = parent_->GetHandleOnDevice(device_name_, *handle); in Instantiate()
758 LocalHandle h = parent_->GetHandleOnDevice(device_name_, handle); in ReleaseHandle()
1063 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, handle); in Run()
1122 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, handle); in Run()
1167 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, handle); in DebugString()
|
| D | process_function_library_runtime.h | 101 FunctionLibraryRuntime::LocalHandle GetHandleOnDevice(
|
| D | process_function_library_runtime.cc | 235 return GetHandleOnDevice(device_name, handle) != kInvalidHandle; in IsInstantiatedOnDevice()
239 ProcessFunctionLibraryRuntime::GetHandleOnDevice( in GetHandleOnDevice() function in tensorflow::ProcessFunctionLibraryRuntime
|