• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Generated file (from: avg_pool_quant8_1.mod.py). Do not edit
CreateModel(Model * model)2 void CreateModel(Model *model) {
3   OperandType type1(Type::INT32, {});
4   OperandType type0(Type::TENSOR_QUANT8_ASYMM, {1, 2, 2, 1}, 0.5f, 0);
5   // Phase 1, operands
6   auto op1 = model->addOperand(&type0);
7   auto cons1 = model->addOperand(&type1);
8   auto pad0 = model->addOperand(&type1);
9   auto act = model->addOperand(&type1);
10   auto op3 = model->addOperand(&type0);
11   // Phase 2, operations
12   static int32_t cons1_init[] = {1};
13   model->setOperandValue(cons1, cons1_init, sizeof(int32_t) * 1);
14   static int32_t pad0_init[] = {0};
15   model->setOperandValue(pad0, pad0_init, sizeof(int32_t) * 1);
16   static int32_t act_init[] = {0};
17   model->setOperandValue(act, act_init, sizeof(int32_t) * 1);
18   model->addOperation(ANEURALNETWORKS_AVERAGE_POOL_2D, {op1, pad0, pad0, pad0, pad0, cons1, cons1, cons1, cons1, act}, {op3});
19   // Phase 3, inputs and outputs
20   model->identifyInputsAndOutputs(
21     {op1},
22     {op3});
23   assert(model->isValid());
24 }
25 
is_ignored(int i)26 bool is_ignored(int i) {
27   static std::set<int> ignore = {};
28   return ignore.find(i) != ignore.end();
29 }
30