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/
Dgraph_rewrite_dataset.cc58 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(lib_); in Optimize()
204 params.function_handle_cache = dataset()->function_handle_cache_.get(); in Initialize()
213 params.function_handle_cache = dataset()->function_handle_cache_.get(); in GetNextInternal()
Dmulti_device_iterator_ops.cc60 function_handle_cache_(std::move(function_handle_cache)) { in MultiDeviceIterator()
99 params.function_handle_cache = function_handle_cache_.get(); in GetNextFromShard()
125 return function_handle_cache_.get(); in function_handle_cache()
360 const std::unique_ptr<FunctionHandleCache> function_handle_cache_; member in tensorflow::data::__anond56f2ce10111::MultiDeviceIterator
Dgraph_rewrite_dataset.h86 std::unique_ptr<FunctionHandleCache> function_handle_cache_ = nullptr; variable
Ddataset_test_base.cc113 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
114 params.function_handle_cache = function_handle_cache_.get(); in CreateIteratorContext()
Ddataset_test_base.h189 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc98 params.function_handle_cache = function_handle_cache_.get(); in MakeIterator()
120 function_handle_cache_ = in Dataset()
Dstandalone.h114 std::unique_ptr<FunctionHandleCache> function_handle_cache_; variable