Home | History | Annotate | Download | only in vts_models 
      1 // Generated code. Do not edit
      2 // Create the model
      3 Model createTestModel() {
      4     const std::vector<Operand> operands = {
      5         {
      6             .type = OperandType::TENSOR_FLOAT32,
      7             .dimensions = {2, 8},
      8             .numberOfConsumers = 1,
      9             .scale = 0.0f,
     10             .zeroPoint = 0,
     11             .lifetime = OperandLifeTime::MODEL_INPUT,
     12             .location = {.poolIndex = 0, .offset = 0, .length = 0},
     13         },
     14         {
     15             .type = OperandType::TENSOR_FLOAT32,
     16             .dimensions = {16, 8},
     17             .numberOfConsumers = 1,
     18             .scale = 0.0f,
     19             .zeroPoint = 0,
     20             .lifetime = OperandLifeTime::MODEL_INPUT,
     21             .location = {.poolIndex = 0, .offset = 0, .length = 0},
     22         },
     23         {
     24             .type = OperandType::TENSOR_FLOAT32,
     25             .dimensions = {16, 16},
     26             .numberOfConsumers = 1,
     27             .scale = 0.0f,
     28             .zeroPoint = 0,
     29             .lifetime = OperandLifeTime::MODEL_INPUT,
     30             .location = {.poolIndex = 0, .offset = 0, .length = 0},
     31         },
     32         {
     33             .type = OperandType::TENSOR_FLOAT32,
     34             .dimensions = {16},
     35             .numberOfConsumers = 1,
     36             .scale = 0.0f,
     37             .zeroPoint = 0,
     38             .lifetime = OperandLifeTime::MODEL_INPUT,
     39             .location = {.poolIndex = 0, .offset = 0, .length = 0},
     40         },
     41         {
     42             .type = OperandType::TENSOR_FLOAT32,
     43             .dimensions = {2, 16},
     44             .numberOfConsumers = 1,
     45             .scale = 0.0f,
     46             .zeroPoint = 0,
     47             .lifetime = OperandLifeTime::MODEL_INPUT,
     48             .location = {.poolIndex = 0, .offset = 0, .length = 0},
     49         },
     50         {
     51             .type = OperandType::INT32,
     52             .dimensions = {},
     53             .numberOfConsumers = 1,
     54             .scale = 0.0f,
     55             .zeroPoint = 0,
     56             .lifetime = OperandLifeTime::CONSTANT_COPY,
     57             .location = {.poolIndex = 0, .offset = 0, .length = 4},
     58         },
     59         {
     60             .type = OperandType::TENSOR_FLOAT32,
     61             .dimensions = {2, 16},
     62             .numberOfConsumers = 0,
     63             .scale = 0.0f,
     64             .zeroPoint = 0,
     65             .lifetime = OperandLifeTime::MODEL_OUTPUT,
     66             .location = {.poolIndex = 0, .offset = 0, .length = 0},
     67         },
     68         {
     69             .type = OperandType::TENSOR_FLOAT32,
     70             .dimensions = {2, 16},
     71             .numberOfConsumers = 0,
     72             .scale = 0.0f,
     73             .zeroPoint = 0,
     74             .lifetime = OperandLifeTime::MODEL_OUTPUT,
     75             .location = {.poolIndex = 0, .offset = 0, .length = 0},
     76         }
     77     };
     78 
     79     const std::vector<Operation> operations = {
     80         {
     81             .type = OperationType::RNN,
     82             .inputs = {0, 1, 2, 3, 4, 5},
     83             .outputs = {6, 7},
     84         }
     85     };
     86 
     87     const std::vector<uint32_t> inputIndexes = {0, 1, 2, 3, 4};
     88     const std::vector<uint32_t> outputIndexes = {6, 7};
     89     std::vector<uint8_t> operandValues = {
     90       1, 0, 0, 0
     91     };
     92     const std::vector<hidl_memory> pools = {};
     93 
     94     return {
     95         .operands = operands,
     96         .operations = operations,
     97         .inputIndexes = inputIndexes,
     98         .outputIndexes = outputIndexes,
     99         .operandValues = operandValues,
    100         .pools = pools,
    101         .relaxComputationFloat32toFloat16 = true,
    102     };
    103 }
    104 
    105 
    106 bool is_ignored(int i) {
    107   static std::set<int> ignore = {0};
    108   return ignore.find(i) != ignore.end();
    109 }
    110