Searched refs:function_handle_cache (Results 1 – 9 of 9) sorted by relevance
80 params.function_handle_cache = in GetNext()81 captured_state->function_handle_cache.get(); in GetNext()168 params.function_handle_cache = new_state->function_handle_cache.get(); in Restore()182 params.function_handle_cache = new_state->function_handle_cache.get(); in Restore()235 new_state->function_handle_cache = in SetIteratorFromDataset()241 params.function_handle_cache = new_state->function_handle_cache.get(); in SetIteratorFromDataset()273 function_handle_cache(absl::make_unique<FunctionHandleCache>(lib)), in State()279 std::unique_ptr<FunctionHandleCache> function_handle_cache, in State()284 function_handle_cache(std::move(function_handle_cache)), in State()290 std::unique_ptr<FunctionHandleCache> function_handle_cache; member[all …]
52 std::unique_ptr<FunctionHandleCache> function_handle_cache) in MultiDeviceIterator() argument60 function_handle_cache_(std::move(function_handle_cache)) { in MultiDeviceIterator()99 params.function_handle_cache = function_handle_cache_.get(); in GetNextFromShard()124 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator410 std::unique_ptr<FunctionHandleCache> function_handle_cache = in Compute() local424 std::move(function_handle_cache)); in Compute()436 &function_handle_cache](MultiDeviceIterator** ret) in Compute()442 lib, std::move(function_handle_cache)); in Compute()509 params.function_handle_cache = resource->function_handle_cache(); in Compute()
204 params.function_handle_cache = dataset()->function_handle_cache_.get(); in Initialize()213 params.function_handle_cache = dataset()->function_handle_cache_.get(); in GetNextInternal()
114 params.function_handle_cache = function_handle_cache_.get(); in CreateIteratorContext()
144 TF_RETURN_IF_ERROR(ctx->function_handle_cache()->Instantiate( in Instantiate()
71 std::unique_ptr<FunctionHandleCache> function_handle_cache = in ComputeAsync() local73 params.function_handle_cache = function_handle_cache.get(); in ComputeAsync()
287 function_handle_cache(ctx->function_handle_cache()), in Params()327 FunctionHandleCache* function_handle_cache = nullptr; member372 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function373 return params_.function_handle_cache; in function_handle_cache()
98 params.function_handle_cache = function_handle_cache_.get(); in MakeIterator()
902 "framework/function_handle_cache.h",