Home | History | Annotate | Download | only in models
      1 // Generated file (from: avg_pool_quant8_3.mod.py). Do not edit
      2 void CreateModel(Model *model) {
      3   OperandType type1(Type::INT32, {});
      4   OperandType type0(Type::TENSOR_QUANT8_ASYMM, {1, 100, 100, 1}, 0.5f, 0);
      5   OperandType type2(Type::TENSOR_QUANT8_ASYMM, {1, 23, 23, 1}, 0.5f, 0);
      6   // Phase 1, operands
      7   auto i0 = model->addOperand(&type0);
      8   auto stride = model->addOperand(&type1);
      9   auto filter = model->addOperand(&type1);
     10   auto padding = model->addOperand(&type1);
     11   auto activation = model->addOperand(&type1);
     12   auto output = model->addOperand(&type2);
     13   // Phase 2, operations
     14   static int32_t stride_init[] = {4};
     15   model->setOperandValue(stride, stride_init, sizeof(int32_t) * 1);
     16   static int32_t filter_init[] = {10};
     17   model->setOperandValue(filter, filter_init, sizeof(int32_t) * 1);
     18   static int32_t padding_init[] = {0};
     19   model->setOperandValue(padding, padding_init, sizeof(int32_t) * 1);
     20   static int32_t activation_init[] = {0};
     21   model->setOperandValue(activation, activation_init, sizeof(int32_t) * 1);
     22   model->addOperation(ANEURALNETWORKS_AVERAGE_POOL_2D, {i0, padding, padding, padding, padding, stride, stride, filter, filter, activation}, {output});
     23   // Phase 3, inputs and outputs
     24   model->identifyInputsAndOutputs(
     25     {i0},
     26     {output});
     27   assert(model->isValid());
     28 }
     29 
     30 bool is_ignored(int i) {
     31   static std::set<int> ignore = {};
     32   return ignore.find(i) != ignore.end();
     33 }
     34