Home
last modified time | relevance | path

Searched refs:FunctionHandleCache (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dfunction_handle_cache.cc24 FunctionHandleCache::FunctionHandleCache(FunctionLibraryRuntime* lib) in FunctionHandleCache() function in tensorflow::data::FunctionHandleCache
29 FunctionHandleCache::~FunctionHandleCache() { in ~FunctionHandleCache()
36 Status FunctionHandleCache::Instantiate( in Instantiate()
58 Status FunctionHandleCache::Clear() { in Clear()
Dfunction_handle_cache.h25 class FunctionHandleCache {
27 explicit FunctionHandleCache(FunctionLibraryRuntime* lib);
29 ~FunctionHandleCache();
Ddataset.h294 class FunctionHandleCache; variable
404 FunctionHandleCache* function_handle_cache = nullptr;
454 FunctionHandleCache* function_handle_cache() { in function_handle_cache()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h85 function_handle_cache_(absl::make_unique<FunctionHandleCache>(flr)), in State()
102 FunctionHandleCache* function_handle_cache() { in function_handle_cache()
118 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
Dmulti_device_iterator_ops.cc61 std::unique_ptr<FunctionHandleCache> function_handle_cache) in MultiDeviceIterator()
141 FunctionHandleCache* function_handle_cache() { in function_handle_cache()
383 const std::unique_ptr<FunctionHandleCache> function_handle_cache_;
434 absl::make_unique<FunctionHandleCache>(flr); in Compute()
529 auto function_handle_cache = absl::make_unique<FunctionHandleCache>(lib); in CreateResource()
Diterator_ops.cc636 FunctionHandleCache function_handle_cache(params.flr); in DoCompute()
708 absl::make_unique<FunctionHandleCache>(params.flr); in DoCompute()
Ddataset_test_base.h765 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
Ddataset_test_base.cc355 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
/external/tensorflow/tensorflow/core/data/
Dstandalone.h120 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
Dstandalone.cc154 absl::make_unique<FunctionHandleCache>(pflr_->GetFLR("/device:CPU:0")); in Dataset()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc78 FunctionHandleCache function_handle_cache(params.flr); in DoCompute()
Dio_ops.cc78 absl::make_unique<FunctionHandleCache>(params.flr); in WriteData()
/external/tensorflow/tensorflow/core/kernels/
Dlookup_util.cc410 absl::make_unique<data::FunctionHandleCache>(params.flr); in Init()
451 std::unique_ptr<data::FunctionHandleCache> function_handle_cache_;