Miao Wang | bee07f7 | 2017-10-09 10:43:21 -0700 | [diff] [blame] | 1 | // Generated file (from: conv_quant8_overflow_weights_as_inputs.mod.py). Do not edit |
| 2 | void CreateModel(Model *model) { |
| 3 | OperandType type3(Type::INT32, {}); |
| 4 | OperandType type2(Type::TENSOR_INT32, {3}, 0.25, 0); |
| 5 | OperandType type0(Type::TENSOR_QUANT8_ASYMM, {1, 2, 3, 3}, 0.5, 0); |
| 6 | OperandType type4(Type::TENSOR_QUANT8_ASYMM, {1, 2, 3, 3}, 1.0, 0); |
| 7 | OperandType type1(Type::TENSOR_QUANT8_ASYMM, {3, 1, 1, 3}, 0.5, 0); |
| 8 | // Phase 1, operands |
| 9 | auto op1 = model->addOperand(&type0); |
| 10 | auto op2 = model->addOperand(&type1); |
| 11 | auto op3 = model->addOperand(&type2); |
| 12 | auto pad0 = model->addOperand(&type3); |
| 13 | auto act = model->addOperand(&type3); |
| 14 | auto stride = model->addOperand(&type3); |
| 15 | auto op4 = model->addOperand(&type4); |
| 16 | // Phase 2, operations |
| 17 | static int32_t pad0_init[] = {0}; |
| 18 | model->setOperandValue(pad0, pad0_init, sizeof(int32_t) * 1); |
| 19 | static int32_t act_init[] = {0}; |
| 20 | model->setOperandValue(act, act_init, sizeof(int32_t) * 1); |
| 21 | static int32_t stride_init[] = {1}; |
| 22 | model->setOperandValue(stride, stride_init, sizeof(int32_t) * 1); |
| 23 | model->addOperation(ANEURALNETWORKS_CONV_2D, {op1, op2, op3, pad0, pad0, pad0, pad0, stride, stride, act}, {op4}); |
| 24 | // Phase 3, inputs and outputs |
| 25 | model->identifyInputsAndOutputs( |
| 26 | {op1, op2, op3}, |
| 27 | {op4}); |
| 28 | assert(model->isValid()); |
| 29 | } |
| 30 | |
| 31 | bool is_ignored(int i) { |
| 32 | static std::set<int> ignore = {}; |
| 33 | return ignore.find(i) != ignore.end(); |
| 34 | } |