mirror of
https://github.com/Evolution-X/hardware_interfaces
synced 2026-02-02 20:24:19 +00:00
72 lines
3.1 KiB
C++
72 lines
3.1 KiB
C++
/*
|
|
* Copyright (C) 2020 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_UTILS_COMMON_COMMON_UTILS_H
|
|
#define ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_UTILS_COMMON_COMMON_UTILS_H
|
|
|
|
#include <cutils/native_handle.h>
|
|
#include <hidl/HidlSupport.h>
|
|
#include <nnapi/Result.h>
|
|
#include <nnapi/Types.h>
|
|
#include <functional>
|
|
#include <vector>
|
|
|
|
// Shorthand
|
|
namespace android::hardware::neuralnetworks {
|
|
namespace hal = ::android::hardware::neuralnetworks;
|
|
} // namespace android::hardware::neuralnetworks
|
|
|
|
// Shorthand
|
|
namespace android::nn {
|
|
namespace hal = ::android::hardware::neuralnetworks;
|
|
}
|
|
|
|
namespace android::hardware::neuralnetworks::utils {
|
|
|
|
nn::Capabilities::OperandPerformanceTable makeQuantized8PerformanceConsistentWithP(
|
|
const nn::Capabilities::PerformanceInfo& float32Performance,
|
|
const nn::Capabilities::PerformanceInfo& quantized8Performance);
|
|
|
|
// Indicates if the object contains no pointer-based data that could be relocated to shared memory.
|
|
bool hasNoPointerData(const nn::Model& model);
|
|
bool hasNoPointerData(const nn::Request& request);
|
|
|
|
// Relocate pointer-based data to shared memory.
|
|
nn::GeneralResult<std::reference_wrapper<const nn::Model>> flushDataFromPointerToShared(
|
|
const nn::Model* model, std::optional<nn::Model>* maybeModelInSharedOut);
|
|
nn::GeneralResult<std::reference_wrapper<const nn::Request>> flushDataFromPointerToShared(
|
|
const nn::Request* request, std::optional<nn::Request>* maybeRequestInSharedOut);
|
|
|
|
// Undoes `flushDataFromPointerToShared` on a Request object. More specifically,
|
|
// `unflushDataFromSharedToPointer` copies the output shared memory data from the transformed
|
|
// Request object back to the output pointer-based memory in the original Request object.
|
|
nn::GeneralResult<void> unflushDataFromSharedToPointer(
|
|
const nn::Request& request, const std::optional<nn::Request>& maybeRequestInShared);
|
|
|
|
std::vector<uint32_t> countNumberOfConsumers(size_t numberOfOperands,
|
|
const std::vector<nn::Operation>& operations);
|
|
|
|
nn::GeneralResult<nn::Memory> createSharedMemoryFromHidlMemory(const hidl_memory& memory);
|
|
|
|
nn::GeneralResult<hidl_handle> hidlHandleFromSharedHandle(const nn::SharedHandle& handle);
|
|
nn::GeneralResult<nn::SharedHandle> sharedHandleFromNativeHandle(const native_handle_t* handle);
|
|
nn::GeneralResult<hidl_vec<hidl_handle>> convertSyncFences(
|
|
const std::vector<nn::SyncFence>& fences);
|
|
|
|
} // namespace android::hardware::neuralnetworks::utils
|
|
|
|
#endif // ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_UTILS_COMMON_COMMON_UTILS_H
|