Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 9e5c9088 authored by TreeHugger Robot's avatar TreeHugger Robot Committed by Android (Google) Code Review
Browse files

Merge "Add FusedActivationFunc enum and renamed certain operations." into oc-mr1-dev

parents 710b7085 199aa864
Loading
Loading
Loading
Loading
+37 −32
Original line number Original line Diff line number Diff line
@@ -44,38 +44,43 @@ enum OperandType : uint32_t {
enum OperationType : uint32_t {
enum OperationType : uint32_t {
    OEM_OPERATION                = 0,
    OEM_OPERATION                = 0,
    ADD                          = 1,
    ADD                          = 1,
    AVERAGE_POOL                 = 2,
    AVERAGE_POOL_2D              = 2,
    CAST                         = 3,
    CONCATENATION                = 3,
    CONCATENATION                = 4,
    CONV_2D                      = 4,
    CONV                         = 5,
    DEPTHWISE_CONV_2D            = 5,
    DEPTHWISE_CONV               = 6,
    DEPTH_TO_SPACE               = 6,
    DEPTH_TO_SPACE               = 7,
    DEQUANTIZE                   = 7,
    DEQUANTIZE                   = 8,
    EMBEDDING_LOOKUP             = 8,
    EMBEDDING_LOOKUP             = 9,
    FAKE_QUANT                   = 9,
    FAKE_QUANT                   = 10,
    FLOOR                        = 10,
    FLOOR                        = 11,
    FULLY_CONNECTED              = 11,
    FULLY_CONNECTED              = 12,
    HASHTABLE_LOOKUP             = 12,
    GATHER                       = 13,
    L2_NORMALIZATION             = 13,
    HASHTABLE_LOOKUP             = 14,
    L2_POOL_2D                   = 14,
    L2_NORMALIZATION             = 15,
    LOCAL_RESPONSE_NORMALIZATION = 15,
    L2_POOL                      = 16,
    LOGISTIC                     = 16,
    LOCAL_RESPONSE_NORMALIZATION = 17,
    LSH_PROJECTION               = 17,
    LOGISTIC                     = 18,
    LSTM                         = 18,
    LSH_PROJECTION               = 19,
    MAX_POOL_2D                  = 19,
    LSTM                         = 20,
    MUL                          = 20,
    MAX_POOL                     = 21,
    RELU                         = 21,
    MUL                          = 22,
    RELU1                        = 22,
    RELU                         = 23,
    RELU6                        = 23,
    RELU1                        = 24,
    RESHAPE                      = 24,
    RELU6                        = 25,
    RESIZE_BILINEAR              = 25,
    RESHAPE                      = 26,
    RNN                          = 26,
    RESIZE_BILINEAR              = 27,
    SOFTMAX                      = 27,
    RNN                          = 28,
    SPACE_TO_DEPTH               = 28,
    SOFTMAX                      = 29,
    SVDF                         = 29,
    SPACE_TO_DEPTH               = 30,
    TANH                         = 30,
    SPLIT                        = 31,
};
    SVDF                         = 32,

    TANH                         = 33,
// 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.
// Two special values that can be used instead of a regular poolIndex.
+1 −1
Original line number Original line Diff line number Diff line
@@ -150,7 +150,7 @@ Model createTestModel() {
    std::vector<uint8_t> operandValues(
    std::vector<uint8_t> operandValues(
        reinterpret_cast<const uint8_t*>(operand2Data.data()),
        reinterpret_cast<const uint8_t*>(operand2Data.data()),
        reinterpret_cast<const uint8_t*>(operand2Data.data()) + size);
        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]),
    operandValues.insert(operandValues.end(), reinterpret_cast<const uint8_t*>(&activation[0]),
                         reinterpret_cast<const uint8_t*>(&activation[1]));
                         reinterpret_cast<const uint8_t*>(&activation[1]));


+1 −0
Original line number Original line Diff line number Diff line
@@ -31,6 +31,7 @@ using ::android::hardware::neuralnetworks::V1_0::IDevice;
using ::android::hardware::neuralnetworks::V1_0::IPreparedModel;
using ::android::hardware::neuralnetworks::V1_0::IPreparedModel;
using ::android::hardware::neuralnetworks::V1_0::Capabilities;
using ::android::hardware::neuralnetworks::V1_0::Capabilities;
using ::android::hardware::neuralnetworks::V1_0::DeviceStatus;
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::Model;
using ::android::hardware::neuralnetworks::V1_0::OperationType;
using ::android::hardware::neuralnetworks::V1_0::OperationType;
using ::android::hardware::neuralnetworks::V1_0::PerformanceInfo;
using ::android::hardware::neuralnetworks::V1_0::PerformanceInfo;