1e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang// Generated file (from: embedding_lookup_relaxed.mod.py). Do not edit 2e8e5d34c4159532eb324df393c2c752a508bced1Miao Wangvoid CreateModel(Model *model) { 3e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang OperandType type1(Type::TENSOR_FLOAT32, {3, 2, 4}); 4e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang OperandType type0(Type::TENSOR_INT32, {3}); 5e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang // Phase 1, operands 6e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang auto index = model->addOperand(&type0); 7e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang auto value = model->addOperand(&type1); 8e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang auto output = model->addOperand(&type1); 9e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang // Phase 2, operations 10e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang model->addOperation(ANEURALNETWORKS_EMBEDDING_LOOKUP, {index, value}, {output}); 11e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang // Phase 3, inputs and outputs 12e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang model->identifyInputsAndOutputs( 13e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang {index, value}, 14e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang {output}); 15e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang // Phase 4: set relaxed execution 16e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang model->relaxComputationFloat32toFloat16(true); 17e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang assert(model->isValid()); 18e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang} 19e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang 20e8e5d34c4159532eb324df393c2c752a508bced1Miao Wangbool is_ignored(int i) { 21e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang static std::set<int> ignore = {}; 22e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang return ignore.find(i) != ignore.end(); 23e8e5d34c4159532eb324df393c2c752a508bced1Miao Wang} 24