Searched refs:GetHandleOnDevice (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/ |
D | function.cc | 571 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, h); in GetFunctionBody() 589 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, h); in GetRetTypes() 758 parent_->GetHandleOnDevice(device_name_, *handle); in Instantiate() 804 local_handle = parent_->GetHandleOnDevice(device_name_, *handle); in Instantiate() 830 LocalHandle h = parent_->GetHandleOnDevice(device_name_, handle); in ReleaseHandle() 1111 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, handle); in Run() 1190 LocalHandle local_handle = parent_->GetHandleOnDevice( in Run() 1253 LocalHandle local_handle = parent_->GetHandleOnDevice( in PrepareRunSync() 1312 LocalHandle local_handle = parent_->GetHandleOnDevice(device_name_, handle); in DebugString()
|
D | process_function_library_runtime_test.cc | 501 EXPECT_EQ(0, proc_flr_->GetHandleOnDevice( in TEST_F() 506 EXPECT_EQ(0, proc_flr_->GetHandleOnDevice( in TEST_F() 512 EXPECT_EQ(1, proc_flr_->GetHandleOnDevice( in TEST_F() 530 EXPECT_EQ(0, proc_flr_->GetHandleOnDevice( in TEST_F()
|
D | process_function_library_runtime.h | 143 FunctionLibraryRuntime::LocalHandle GetHandleOnDevice(
|
D | process_function_library_runtime.cc | 288 return GetHandleOnDevice(device_name, handle) != kInvalidHandle; in IsInstantiatedOnDevice() 292 ProcessFunctionLibraryRuntime::GetHandleOnDevice( in GetHandleOnDevice() function in tensorflow::ProcessFunctionLibraryRuntime
|