mirror of
https://github.com/Evolution-X/hardware_interfaces
synced 2026-02-01 16:50:18 +00:00
Merge "Add FusedActivationFunc enum and renamed certain operations." into oc-mr1-dev am: 9e5c908810
am: 3c7df7d717
Change-Id: I8e9ff6601c2bf2738ce86c7937e83366096d909f
This commit is contained in:
@@ -44,38 +44,43 @@ enum OperandType : uint32_t {
|
||||
enum OperationType : uint32_t {
|
||||
OEM_OPERATION = 0,
|
||||
ADD = 1,
|
||||
AVERAGE_POOL = 2,
|
||||
CAST = 3,
|
||||
CONCATENATION = 4,
|
||||
CONV = 5,
|
||||
DEPTHWISE_CONV = 6,
|
||||
DEPTH_TO_SPACE = 7,
|
||||
DEQUANTIZE = 8,
|
||||
EMBEDDING_LOOKUP = 9,
|
||||
FAKE_QUANT = 10,
|
||||
FLOOR = 11,
|
||||
FULLY_CONNECTED = 12,
|
||||
GATHER = 13,
|
||||
HASHTABLE_LOOKUP = 14,
|
||||
L2_NORMALIZATION = 15,
|
||||
L2_POOL = 16,
|
||||
LOCAL_RESPONSE_NORMALIZATION = 17,
|
||||
LOGISTIC = 18,
|
||||
LSH_PROJECTION = 19,
|
||||
LSTM = 20,
|
||||
MAX_POOL = 21,
|
||||
MUL = 22,
|
||||
RELU = 23,
|
||||
RELU1 = 24,
|
||||
RELU6 = 25,
|
||||
RESHAPE = 26,
|
||||
RESIZE_BILINEAR = 27,
|
||||
RNN = 28,
|
||||
SOFTMAX = 29,
|
||||
SPACE_TO_DEPTH = 30,
|
||||
SPLIT = 31,
|
||||
SVDF = 32,
|
||||
TANH = 33,
|
||||
AVERAGE_POOL_2D = 2,
|
||||
CONCATENATION = 3,
|
||||
CONV_2D = 4,
|
||||
DEPTHWISE_CONV_2D = 5,
|
||||
DEPTH_TO_SPACE = 6,
|
||||
DEQUANTIZE = 7,
|
||||
EMBEDDING_LOOKUP = 8,
|
||||
FAKE_QUANT = 9,
|
||||
FLOOR = 10,
|
||||
FULLY_CONNECTED = 11,
|
||||
HASHTABLE_LOOKUP = 12,
|
||||
L2_NORMALIZATION = 13,
|
||||
L2_POOL_2D = 14,
|
||||
LOCAL_RESPONSE_NORMALIZATION = 15,
|
||||
LOGISTIC = 16,
|
||||
LSH_PROJECTION = 17,
|
||||
LSTM = 18,
|
||||
MAX_POOL_2D = 19,
|
||||
MUL = 20,
|
||||
RELU = 21,
|
||||
RELU1 = 22,
|
||||
RELU6 = 23,
|
||||
RESHAPE = 24,
|
||||
RESIZE_BILINEAR = 25,
|
||||
RNN = 26,
|
||||
SOFTMAX = 27,
|
||||
SPACE_TO_DEPTH = 28,
|
||||
SVDF = 29,
|
||||
TANH = 30,
|
||||
};
|
||||
|
||||
// Fused activation functions
|
||||
enum FusedActivationFunc : int32_t {
|
||||
NONE = 0,
|
||||
RELU = 1,
|
||||
RELU1 = 2,
|
||||
RELU6 = 3,
|
||||
};
|
||||
|
||||
// Two special values that can be used instead of a regular poolIndex.
|
||||
|
||||
@@ -150,7 +150,7 @@ Model createTestModel() {
|
||||
std::vector<uint8_t> operandValues(
|
||||
reinterpret_cast<const uint8_t*>(operand2Data.data()),
|
||||
reinterpret_cast<const uint8_t*>(operand2Data.data()) + size);
|
||||
int32_t activation[1] = {0};
|
||||
int32_t activation[1] = {static_cast<int32_t>(FusedActivationFunc::NONE)};
|
||||
operandValues.insert(operandValues.end(), reinterpret_cast<const uint8_t*>(&activation[0]),
|
||||
reinterpret_cast<const uint8_t*>(&activation[1]));
|
||||
|
||||
|
||||
@@ -31,6 +31,7 @@ using ::android::hardware::neuralnetworks::V1_0::IDevice;
|
||||
using ::android::hardware::neuralnetworks::V1_0::IPreparedModel;
|
||||
using ::android::hardware::neuralnetworks::V1_0::Capabilities;
|
||||
using ::android::hardware::neuralnetworks::V1_0::DeviceStatus;
|
||||
using ::android::hardware::neuralnetworks::V1_0::FusedActivationFunc;
|
||||
using ::android::hardware::neuralnetworks::V1_0::Model;
|
||||
using ::android::hardware::neuralnetworks::V1_0::OperationType;
|
||||
using ::android::hardware::neuralnetworks::V1_0::PerformanceInfo;
|
||||
|
||||
Reference in New Issue
Block a user