Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h85 function_handle_cache_(absl::make_unique<FunctionHandleCache>(flr)), in State()
103 return function_handle_cache_.get(); in function_handle_cache()
118 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable
Dmulti_device_iterator_ops.cc69 function_handle_cache_(std::move(function_handle_cache)) { in MultiDeviceIterator()
107 params.function_handle_cache = function_handle_cache_.get(); in GetNextFromShard()
142 return function_handle_cache_.get(); in function_handle_cache()
383 const std::unique_ptr<FunctionHandleCache> function_handle_cache_; member in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
Ddataset_test_base.cc355 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
356 params.function_handle_cache = function_handle_cache_.get(); in CreateIteratorContext()
Ddataset_test_base.h765 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable
/external/tensorflow/tensorflow/core/data/
Dstandalone.cc116 params.function_handle_cache = function_handle_cache_.get(); in MakeIterator()
153 function_handle_cache_ = in Dataset()
Dstandalone.h120 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable
/external/tensorflow/tensorflow/core/kernels/
Dlookup_util.cc409 function_handle_cache_ = in Init()
411 params.function_handle_cache = function_handle_cache_.get(); in Init()
451 std::unique_ptr<data::FunctionHandleCache> function_handle_cache_; member in tensorflow::lookup::DatasetIterator