Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dfunction_handle_cache.cc23 FunctionHandleCache::FunctionHandleCache(FunctionLibraryRuntime* lib) in FunctionHandleCache() function in tensorflow::FunctionHandleCache
28 FunctionHandleCache::~FunctionHandleCache() { in ~FunctionHandleCache()
35 Status FunctionHandleCache::Instantiate( in Instantiate()
57 Status FunctionHandleCache::Clear() { in Clear()
Dfunction_handle_cache.h25 class FunctionHandleCache {
27 explicit FunctionHandleCache(FunctionLibraryRuntime* lib);
29 ~FunctionHandleCache();
Ddataset.h459 FunctionHandleCache* function_handle_cache = nullptr;
528 FunctionHandleCache* function_handle_cache() { in function_handle_cache()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h90 function_handle_cache_(std::make_unique<FunctionHandleCache>(flr)), in State()
112 FunctionHandleCache* function_handle_cache() { in function_handle_cache()
130 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
Dmulti_device_iterator_ops.cc84 std::unique_ptr<FunctionHandleCache> function_handle_cache) in MultiDeviceIterator()
161 FunctionHandleCache* function_handle_cache() { in function_handle_cache()
467 const std::unique_ptr<FunctionHandleCache> function_handle_cache_;
518 auto function_handle_cache = std::make_unique<FunctionHandleCache>(flr); in Compute()
618 auto function_handle_cache = std::make_unique<FunctionHandleCache>(lib); in CreateResource()
Dreduce_dataset_op.cc65 std::make_unique<FunctionHandleCache>(params.flr); in DoCompute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dlookup_ops.cc58 function_handle_cache_ = std::make_unique<FunctionHandleCache>(params.flr); in Init()
104 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
Dto_tf_record_op.cc80 FunctionHandleCache function_handle_cache(params.flr); in DoCompute()
Dio_ops.cc111 std::make_unique<FunctionHandleCache>(params.flr); in WriteData()
/external/tensorflow/tensorflow/core/data/
Dstandalone.h126 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
Dstandalone.cc182 std::make_unique<FunctionHandleCache>(pflr_->GetFLR("/device:CPU:0")); in Dataset()
Dcaptured_function.h144 FunctionHandleCache* function_handle_cache;
Ddataset_test_base.h818 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
Ddataset_test_base.cc360 function_handle_cache_ = std::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc430 std::unique_ptr<FunctionHandleCache> function_handle_cache_;
488 function_handle_cache_(std::make_unique<FunctionHandleCache>(this)), in FunctionLibraryRuntimeImpl()