Searched refs:FunctionHandleCache (Results 1 – 12 of 12) sorted by relevance
24 FunctionHandleCache::FunctionHandleCache(FunctionLibraryRuntime* lib) in FunctionHandleCache() function in tensorflow::data::FunctionHandleCache27 FunctionHandleCache::~FunctionHandleCache() { in ~FunctionHandleCache()34 Status FunctionHandleCache::Instantiate( in Instantiate()56 Status FunctionHandleCache::Clear() { in Clear()
25 class FunctionHandleCache {27 explicit FunctionHandleCache(FunctionLibraryRuntime* lib);29 ~FunctionHandleCache();
266 class FunctionHandleCache; variable327 FunctionHandleCache* function_handle_cache = nullptr;372 FunctionHandleCache* function_handle_cache() { in function_handle_cache()
71 std::unique_ptr<FunctionHandleCache> function_handle_cache = in ComputeAsync()72 absl::make_unique<FunctionHandleCache>(params.lib); in ComputeAsync()
52 std::unique_ptr<FunctionHandleCache> function_handle_cache) in MultiDeviceIterator()124 FunctionHandleCache* function_handle_cache() { in function_handle_cache()360 const std::unique_ptr<FunctionHandleCache> function_handle_cache_;410 std::unique_ptr<FunctionHandleCache> function_handle_cache = in Compute()411 absl::make_unique<FunctionHandleCache>(lib); in Compute()
236 absl::make_unique<FunctionHandleCache>(new_state->lib); in SetIteratorFromDataset()273 function_handle_cache(absl::make_unique<FunctionHandleCache>(lib)), in State()279 std::unique_ptr<FunctionHandleCache> function_handle_cache, in State()290 std::unique_ptr<FunctionHandleCache> function_handle_cache;580 std::unique_ptr<FunctionHandleCache> function_handle_cache = in ComputeAsync()581 absl::make_unique<FunctionHandleCache>(params.lib); in ComputeAsync()671 std::unique_ptr<FunctionHandleCache> function_handle_cache = in ComputeAsync()672 absl::make_unique<FunctionHandleCache>(params.lib); in ComputeAsync()
86 std::unique_ptr<FunctionHandleCache> function_handle_cache_ = nullptr;
189 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
58 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(lib_); in Optimize()
113 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
114 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
121 MakeUnique<FunctionHandleCache>(pflr_->GetFLR("/device:CPU:0")); in Dataset()