Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc84 std::unique_ptr<FunctionHandleCache> function_handle_cache) in MultiDeviceIterator() argument
94 function_handle_cache_(std::move(function_handle_cache)) { in MultiDeviceIterator()
139 params.function_handle_cache = function_handle_cache_.get(); in GetNextFromShard()
161 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function in tensorflow::data::__anon2efe57860111::MultiDeviceIterator
518 auto function_handle_cache = std::make_unique<FunctionHandleCache>(flr); in Compute() local
531 std::move(function_handle_cache)); in Compute()
543 &function_handle_cache](MultiDeviceIterator** ret) in Compute()
549 flr, std::move(function_handle_cache)); in Compute()
618 auto function_handle_cache = std::make_unique<FunctionHandleCache>(lib); in CreateResource() local
622 lib, std::move(function_handle_cache)); in CreateResource()
[all …]
Dreduce_dataset_op.cc64 auto function_handle_cache = in DoCompute() local
66 params.function_handle_cache = function_handle_cache.get(); in DoCompute()
Diterator_ops.cc127 params.function_handle_cache = captured_state->function_handle_cache(); in GetNext()
191 params.function_handle_cache = new_state->function_handle_cache(); in Restore()
227 params.function_handle_cache = new_state->function_handle_cache(); in SetIteratorFromDataset()
Diterator_ops.h112 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc80 FunctionHandleCache function_handle_cache(params.flr); in DoCompute() local
81 params.function_handle_cache = &function_handle_cache; in DoCompute()
Dio_ops.cc110 auto function_handle_cache = in WriteData() local
112 params.function_handle_cache = function_handle_cache.get(); in WriteData()
Dlookup_ops.cc59 params.function_handle_cache = function_handle_cache_.get(); in Init()
/external/tensorflow/tensorflow/core/data/
Dcaptured_function.h133 function_handle_cache = ctx->function_handle_cache(); in InstantiateCapturedFunctionParams()
139 function_handle_cache = nullptr; in InstantiateCapturedFunctionParams()
144 FunctionHandleCache* function_handle_cache; member
Dstandalone.cc139 params.function_handle_cache = function_handle_cache_.get(); in MakeIterator()
Dcaptured_function.cc555 if (!params.function_handle_cache) { in Instantiate()
661 if (params.function_handle_cache) { in Instantiate()
662 TF_RETURN_IF_ERROR(params.function_handle_cache->Instantiate( in Instantiate()
Ddataset_test_base.cc361 params.function_handle_cache = function_handle_cache_.get(); in CreateIteratorContext()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h401 function_handle_cache(ctx->function_handle_cache()), in Params()
459 FunctionHandleCache* function_handle_cache = nullptr; member
528 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function
529 return params_.function_handle_cache; in function_handle_cache()
DAndroid.bp40 "function_handle_cache.cc",
DBUILD57 "function_handle_cache.h",
193 "function_handle_cache.h",
270 "function_handle_cache.cc",
373 "function_handle_cache.cc",
374 "function_handle_cache.h",
/external/tensorflow/tensorflow/core/
DBUILD446 "//tensorflow/core/framework:function_handle_cache.h",