Home
last modified time | relevance | path

Searched refs:tensor_memory_map (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc2187 tensor_memory_map) { in CreateExecutionCacheSignature()
2192 if (handle < 0 || handle >= tensor_memory_map.size()) { in CreateExecutionCacheSignature()
2195 tensor_handle_timestamps[i] = tensor_memory_map[handle].timestamp; in CreateExecutionCacheSignature()
6493 return delegate_data->tensor_memory_map; in GetTensorMemoryMap()
6506 int map_size = delegate_data_.tensor_memory_map.size(); in RegisterNnapiMemory()
6508 if (delegate_data_.tensor_memory_map[i].memory == nullptr) { in RegisterNnapiMemory()
6509 delegate_data_.tensor_memory_map[i] = {memory, callback, callback_context, in RegisterNnapiMemory()
6514 delegate_data_.tensor_memory_map.push_back( in RegisterNnapiMemory()
6524 buffer_handle >= delegate_data->tensor_memory_map.size()) { in DoCopyFromBufferHandle()
6527 auto memory = delegate_data->tensor_memory_map[buffer_handle].memory; in DoCopyFromBufferHandle()
[all …]
H A Dnnapi_delegate.h308 std::vector<MemoryRegistration> tensor_memory_map; member