Searched refs:function_handle_cache (Results 1 – 12 of 12) sorted by relevance
78 FunctionHandleCache function_handle_cache(params.flr); in DoCompute() local79 params.function_handle_cache = &function_handle_cache; in DoCompute()
77 auto function_handle_cache = in WriteData() local79 params.function_handle_cache = function_handle_cache.get(); in WriteData()
61 std::unique_ptr<FunctionHandleCache> function_handle_cache) in MultiDeviceIterator() argument69 function_handle_cache_(std::move(function_handle_cache)) { in MultiDeviceIterator()107 params.function_handle_cache = function_handle_cache_.get(); in GetNextFromShard()141 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator433 auto function_handle_cache = in Compute() local447 std::move(function_handle_cache)); in Compute()459 &function_handle_cache](MultiDeviceIterator** ret) in Compute()465 flr, std::move(function_handle_cache)); in Compute()529 auto function_handle_cache = absl::make_unique<FunctionHandleCache>(lib); in CreateResource() local533 lib, std::move(function_handle_cache)); in CreateResource()[all …]
120 params.function_handle_cache = captured_state->function_handle_cache(); in GetNext()208 params.function_handle_cache = new_state->function_handle_cache(); in Restore()243 params.function_handle_cache = new_state->function_handle_cache(); in SetIteratorFromDataset()636 FunctionHandleCache function_handle_cache(params.flr); in DoCompute() local637 params.function_handle_cache = &function_handle_cache; in DoCompute()707 auto function_handle_cache = in DoCompute() local709 params.function_handle_cache = function_handle_cache.get(); in DoCompute()
102 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function
705 TF_RETURN_IF_ERROR(ctx->function_handle_cache()->Instantiate( in Instantiate()
356 params.function_handle_cache = function_handle_cache_.get(); in CreateIteratorContext()
346 function_handle_cache(ctx->function_handle_cache()), in Params()404 FunctionHandleCache* function_handle_cache = nullptr; member454 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function455 return params_.function_handle_cache; in function_handle_cache()
59 "function_handle_cache.h",191 "function_handle_cache.h",266 "function_handle_cache.cc",362 "function_handle_cache.cc",363 "function_handle_cache.h",
116 params.function_handle_cache = function_handle_cache_.get(); in MakeIterator()
411 params.function_handle_cache = function_handle_cache_.get(); in Init()
425 "//tensorflow/core/framework:function_handle_cache.h",