Home
last modified time | relevance | path

Searched refs:function_handle_cache_ (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dlookup_ops.cc58 function_handle_cache_ = std::make_unique<FunctionHandleCache>(params.flr); in Init()
59 params.function_handle_cache = function_handle_cache_.get(); in Init()
104 std::unique_ptr<FunctionHandleCache> function_handle_cache_; member in tensorflow::data::experimental::__anond85f2e6b0111::DatasetIterator
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Diterator_ops.h90 function_handle_cache_(std::make_unique<FunctionHandleCache>(flr)), in State()
113 return function_handle_cache_.get(); in function_handle_cache()
130 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable
H A Dmulti_device_iterator_ops.cc94 function_handle_cache_(std::move(function_handle_cache)) { in MultiDeviceIterator()
139 params.function_handle_cache = function_handle_cache_.get(); in GetNextFromShard()
162 return function_handle_cache_.get(); in function_handle_cache()
467 const std::unique_ptr<FunctionHandleCache> function_handle_cache_; member in tensorflow::data::__anon7578bfd70111::MultiDeviceIterator
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dstandalone.cc139 params.function_handle_cache = function_handle_cache_.get(); in MakeIterator()
181 function_handle_cache_ = in Dataset()
H A Dstandalone.h126 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable
H A Ddataset_test_base.cc360 function_handle_cache_ = std::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
361 params.function_handle_cache = function_handle_cache_.get(); in CreateIteratorContext()
H A Ddataset_test_base.h818 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dfunction.cc430 std::unique_ptr<FunctionHandleCache> function_handle_cache_; member in tensorflow::FunctionLibraryRuntimeImpl
488 function_handle_cache_(std::make_unique<FunctionHandleCache>(this)), in FunctionLibraryRuntimeImpl()
755 return function_handle_cache_->Instantiate(function_name, attrs, in Instantiate()