Searched refs:tensor_memory_map (Results 1 – 2 of 2) sorted by relevance
226 std::vector<MemoryRegistration> tensor_memory_map; member
4893 return delegate_data->tensor_memory_map; in GetTensorMemoryMap()4899 int map_size = delegate_data_.tensor_memory_map.size(); in RegisterNnapiMemory()4901 if (delegate_data_.tensor_memory_map[i].memory == nullptr) { in RegisterNnapiMemory()4902 delegate_data_.tensor_memory_map[i] = {memory, callback, in RegisterNnapiMemory()4907 delegate_data_.tensor_memory_map.push_back( in RegisterNnapiMemory()4917 buffer_handle >= delegate_data->tensor_memory_map.size()) { in DoCopyFromBufferHandle()4920 auto memory = delegate_data->tensor_memory_map[buffer_handle].memory; in DoCopyFromBufferHandle()4921 auto callback = delegate_data->tensor_memory_map[buffer_handle].callback; in DoCopyFromBufferHandle()4923 delegate_data->tensor_memory_map[buffer_handle].callback_context; in DoCopyFromBufferHandle()4940 if (*handle >= 0 && *handle < delegate_data->tensor_memory_map.size()) { in DoFreeBufferHandle()[all …]