Searched refs:GetEntryForKernel (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/ |
H A D | serialization_test.cc | 148 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 D | serialization.h | 185 SerializationEntry GetEntryForKernel( in GetEntryForKernel() function
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/ |
H A D | delegate.cc | 412 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 D | nnapi_delegate.cc | 4554 auto partition_entry = cache->GetEntryForKernel(kNnapiId, context, params); in Init()
|