Searched refs:CachedFunctionHandles (Results 1 – 4 of 4) sorted by relevance
148 class CachedFunctionHandles {150 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles() function162 ~CachedFunctionHandles() { ReleaseAllHandles().IgnoreError(); } in ~CachedFunctionHandles()168 TF_DISALLOW_COPY_AND_ASSIGN(CachedFunctionHandles);
280 TEST(CachedFunctionHandles, Basic) { in TEST() argument304 CachedFunctionHandles cached_function_handles(flr); in TEST()
861 Status CachedFunctionHandles::GetOrInstantiate( in GetOrInstantiate()876 Status CachedFunctionHandles::ReleaseAllHandles() { in ReleaseAllHandles()
2038 CachedFunctionHandles* cached_function_handles, in InferXlaShardingFromNeighbors()2186 CachedFunctionHandles cached_function_handles(flr); in AssignArgsAndRetvalsToCores()