Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A 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::__anon7578bfd70111::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 …]
H A Dreduce_dataset_op.cc64 auto function_handle_cache = in DoCompute() local
66 params.function_handle_cache = function_handle_cache.get(); in DoCompute()
H A 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()
H A Diterator_ops.h112 FunctionHandleCache* function_handle_cache() { in function_handle_cache() function
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A 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
H A 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()
H A Dstandalone.cc139 params.function_handle_cache = function_handle_cache_.get(); in MakeIterator()
H A Ddataset_test_base.cc361 params.function_handle_cache = function_handle_cache_.get(); in CreateIteratorContext()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dto_tf_record_op.cc80 FunctionHandleCache function_handle_cache(params.flr); in DoCompute() local
81 params.function_handle_cache = &function_handle_cache; in DoCompute()
H A Dio_ops.cc110 auto function_handle_cache = in WriteData() local
112 params.function_handle_cache = function_handle_cache.get(); in WriteData()
H A Dlookup_ops.cc59 params.function_handle_cache = function_handle_cache_.get(); in Init()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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()
H A 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",
H A DAndroid.bp40 "function_handle_cache.cc",
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/framework/
Dfunction_handle_cache.o.d1 …rtex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/framework/function_handle_cache.o: \
2 external/tensorflow/tensorflow/core/framework/function_handle_cache.cc \
3 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/
Dmeta_lic181 sources: "external/tensorflow/tensorflow/core/framework/function_handle_cache.cc"
Dmeta_lic.rsp1 …work/function.cc -s external/tensorflow/tensorflow/core/framework/function_handle_cache.cc -s exte…
/aosp_15_r20/external/tensorflow/tensorflow/core/
H A DBUILD446 "//tensorflow/core/framework:function_handle_cache.h",
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/data/
Dserialization_utils.o.d1397 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \
Dunbounded_thread_pool.o.d1401 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \
Ddataset_utils.o.d1423 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.o.d1424 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \
Dmap_dataset_op.o.d1398 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \
Dtake_dataset_op.o.d1397 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/core/kernels/
Dlookup_util.o.d1382 external/tensorflow/tensorflow/core/framework/function_handle_cache.h \

12