1 /* 2 * Copyright (C) 2021 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef ANDROID_FRAMEWORKS_ML_NN_COMMON_AIDL_HAL_UTILS_H 18 #define ANDROID_FRAMEWORKS_ML_NN_COMMON_AIDL_HAL_UTILS_H 19 20 #include <vector> 21 22 #include "AidlHalInterfaces.h" 23 24 namespace android { 25 namespace nn { 26 27 // Return a vector with one entry for each non-extension OperandType except 28 // SUBGRAPH, set to the specified PerformanceInfo value. The vector will be 29 // sorted by OperandType. 30 // 31 // Control flow (OperandType::SUBGRAPH) operation performance is specified 32 // separately using Capabilities::ifPerformance and 33 // Capabilities::whilePerformance. 34 std::vector<aidl_hal::OperandPerformance> nonExtensionOperandPerformance( 35 aidl_hal::PerformanceInfo perf); 36 37 // Update the vector entry corresponding to the specified OperandType with the 38 // specified PerformanceInfo value. The vector must already have an entry for 39 // that OperandType, and must be sorted by OperandType. 40 void update(std::vector<aidl_hal::OperandPerformance>* operandPerformance, 41 aidl_hal::OperandType type, aidl_hal::PerformanceInfo perf); 42 43 // Returns true if an operand type is an extension type. 44 bool isExtensionOperandType(aidl_hal::OperandType type); 45 46 aidl_hal::ErrorStatus convertResultCodeToAidlErrorStatus(int resultCode); 47 48 } // namespace nn 49 } // namespace android 50 51 #endif // ANDROID_FRAMEWORKS_ML_NN_COMMON_AIDL_HAL_UTILS_H 52