hashtable_lookup_float.model.cpp revision ec8e1c71d11f7b5abe306e0c9c6af7da755bd46b
1ec8e1c71d11f7b5abe306e0c9c6af7da755bd46bI-Jui (Ray) Sung// Generated file (from: hashtable_lookup_float.mod.py). Do not edit 2608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sungvoid CreateModel(Model *model) { 3608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung OperandType type2(Type::TENSOR_FLOAT32, {3, 2}); 4608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung OperandType type1(Type::TENSOR_FLOAT32, {3}); 5608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung OperandType type3(Type::TENSOR_FLOAT32, {4, 2}); 6608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung OperandType type0(Type::TENSOR_FLOAT32, {4}); 7608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung // Phase 1, operands 8608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung auto lookup = model->addOperand(&type0); 9608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung auto key = model->addOperand(&type1); 10608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung auto value = model->addOperand(&type2); 11608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung auto output = model->addOperand(&type3); 12608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung auto hits = model->addOperand(&type0); 13608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung // Phase 2, operations 14608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung model->addOperation(ANEURALNETWORKS_HASHTABLE_LOOKUP, {lookup, key, value}, {output}); 15608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung // Phase 3, inputs and outputs 16608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung model->setInputsAndOutputs( 17608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung {lookup, key, value}, 18608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung {output, hits}); 19608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung assert(model->isValid()); 20608124c8a4bc3a5385694d74f0b2f2a6ce5d6292I-Jui (Ray) Sung} 21ec8e1c71d11f7b5abe306e0c9c6af7da755bd46bI-Jui (Ray) Sung 22ec8e1c71d11f7b5abe306e0c9c6af7da755bd46bI-Jui (Ray) Sungbool is_ignored(int i) { 23ec8e1c71d11f7b5abe306e0c9c6af7da755bd46bI-Jui (Ray) Sung static std::set<int> ignore = {}; 24ec8e1c71d11f7b5abe306e0c9c6af7da755bd46bI-Jui (Ray) Sung return ignore.find(i) != ignore.end(); 25ec8e1c71d11f7b5abe306e0c9c6af7da755bd46bI-Jui (Ray) Sung} 26