Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel_test.cc69 const std::map<int, int>& GetTensorReleaseMap(DelegateKernel* kernel) { in GetTensorReleaseMap() function in tflite::flex::testing::KernelTest
70 return kernel->GetTensorReleaseMap(); in GetTensorReleaseMap()
145 const auto& tensor_release_map = GetTensorReleaseMap(delegate_kernel); in TEST_F()
H A Dkernel.h60 const std::map<int, int>& GetTensorReleaseMap() const;
H A Ddelegate_data.h73 std::map<int, int>* GetTensorReleaseMap(const TfLiteContext* context) { in GetTensorReleaseMap() function
H A Dkernel.cc451 flex_delegate_data->GetTensorReleaseMap(context); in Init()
805 const std::map<int, int>& DelegateKernel::GetTensorReleaseMap() const { in GetTensorReleaseMap() function in tflite::flex::DelegateKernel