Searched refs:cached_function_handles (Results 1 – 2 of 2) sorted by relevance
304 CachedFunctionHandles cached_function_handles(flr); in TEST() local312 TF_ASSERT_OK(cached_function_handles.GetOrInstantiate( in TEST()322 TF_ASSERT_OK(cached_function_handles.GetOrInstantiate( in TEST()331 TF_ASSERT_OK(cached_function_handles.GetOrInstantiate( in TEST()336 TF_EXPECT_OK(cached_function_handles.ReleaseAllHandles()); in TEST()
2038 CachedFunctionHandles* cached_function_handles, in InferXlaShardingFromNeighbors() argument2065 TF_RETURN_IF_ERROR(cached_function_handles->GetOrInstantiate( in InferXlaShardingFromNeighbors()2186 CachedFunctionHandles cached_function_handles(flr); in AssignArgsAndRetvalsToCores() local2207 *n, num_cores_per_replica, flr, &cached_function_handles, in AssignArgsAndRetvalsToCores()2336 TF_RETURN_IF_ERROR(cached_function_handles.ReleaseAllHandles()); in AssignArgsAndRetvalsToCores()