Searched refs:CachedFunctionHandles (Results 1 – 4 of 4) sorted by relevance
150 class CachedFunctionHandles {152 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles() function164 ~CachedFunctionHandles() { ReleaseAllHandles().IgnoreError(); } in ~CachedFunctionHandles()170 TF_DISALLOW_COPY_AND_ASSIGN(CachedFunctionHandles);
280 TEST(CachedFunctionHandles, Basic) { in TEST() argument304 CachedFunctionHandles cached_function_handles(flr); in TEST()
687 Status CachedFunctionHandles::GetOrInstantiate( in GetOrInstantiate()702 Status CachedFunctionHandles::ReleaseAllHandles() { in ReleaseAllHandles()
1825 CachedFunctionHandles* cached_function_handles, in InferXlaShardingFromNeighbors()1973 CachedFunctionHandles cached_function_handles(flr); in AssignArgsAndRetvalsToCores()