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()
1825 CachedFunctionHandles* cached_function_handles, in InferXlaShardingFromNeighbors() argument1852 TF_RETURN_IF_ERROR(cached_function_handles->GetOrInstantiate( in InferXlaShardingFromNeighbors()1973 CachedFunctionHandles cached_function_handles(flr); in AssignArgsAndRetvalsToCores() local1993 *n, num_cores_per_replica, flr, &cached_function_handles, in AssignArgsAndRetvalsToCores()2112 TF_RETURN_IF_ERROR(cached_function_handles.ReleaseAllHandles()); in AssignArgsAndRetvalsToCores()