Miao Wang | e8e5d34 | 2018-02-22 19:05:20 -0800 | [diff] [blame] | 1 | // Generated file (from: max_pool_float_1_relaxed.mod.py). Do not edit |
| 2 | void CreateModel(Model *model) { |
| 3 | OperandType type1(Type::INT32, {}); |
| 4 | OperandType type0(Type::TENSOR_FLOAT32, {1, 2, 2, 1}); |
| 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_MAX_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 | // Phase 4: set relaxed execution |
| 24 | model->relaxComputationFloat32toFloat16(true); |
| 25 | assert(model->isValid()); |
| 26 | } |
| 27 | |
| 28 | bool is_ignored(int i) { |
| 29 | static std::set<int> ignore = {}; |
| 30 | return ignore.find(i) != ignore.end(); |
| 31 | } |