| 3325 | } |
| 3326 | |
| 3327 | TfLiteBufferHandle StatefulNnApiDelegate::RegisterNnapiMemory( |
| 3328 | ANeuralNetworksMemory* memory, CopyToHostTensorFnPtr callback, |
| 3329 | void* callback_context) { |
| 3330 | int map_size = delegate_data_.tensor_memory_map.size(); |
| 3331 | for (int i = 0; i < map_size; i++) { |
| 3332 | if (delegate_data_.tensor_memory_map[i].memory == nullptr) { |
| 3333 | delegate_data_.tensor_memory_map[i] = {memory, callback, |
| 3334 | callback_context}; |
| 3335 | return i; |
| 3336 | } |
| 3337 | } |
| 3338 | delegate_data_.tensor_memory_map.push_back( |
| 3339 | {memory, callback, callback_context}); |
| 3340 | return map_size; |
| 3341 | } |
| 3342 | |
| 3343 | TfLiteStatus StatefulNnApiDelegate::DoCopyFromBufferHandle( |
| 3344 | TfLiteContext* context, TfLiteDelegate* delegate, |