Home
last modified time | relevance | path

Searched refs:GetEntryForKernel (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dserialization_test.cc148 auto ref_entry = serialization.GetEntryForKernel( in TEST_F()
156 .GetEntryForKernel(delegate.c_str(), &ref_context, in TEST_F()
165 .GetEntryForKernel(delegate.c_str(), &ref_context, in TEST_F()
174 .GetEntryForKernel(delegate.c_str(), &ref_context, in TEST_F()
183 .GetEntryForKernel(delegate.c_str(), &other_context, &ref_partition) in TEST_F()
190 .GetEntryForKernel(delegate.c_str(), &ref_context, &ref_partition) in TEST_F()
198 .GetEntryForKernel(delegate.c_str(), &ref_context, &ref_partition) in TEST_F()
215 serialization1.GetEntryForKernel(delegate.c_str(), &context, &partition); in TEST_F()
220 serialization2.GetEntryForKernel(delegate.c_str(), &context, &partition); in TEST_F()
231 serialization3.GetEntryForKernel(delegate.c_str(), &context, &partition); in TEST_F()
[all …]
H A Dserialization.h185 SerializationEntry GetEntryForKernel( in GetEntryForKernel() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Ddelegate.cc412 auto data_key = serialization->GetEntryForKernel( in MaybeInitializeSerializedOpenCL()
446 auto data_key = serialization->GetEntryForKernel( in SaveSerializedOpenCL()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc4554 auto partition_entry = cache->GetEntryForKernel(kNnapiId, context, params); in Init()