14d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang// Generated file (from: batch_to_space.mod.py). Do not edit 24d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wangvoid CreateModel(Model *model) { 3a9ce0001ca1df32fde3f322c385eef8ce7279329Miao Wang OperandType type2(Type::TENSOR_FLOAT32, {1, 2, 2, 2}); 44d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang OperandType type0(Type::TENSOR_FLOAT32, {4, 1, 1, 2}); 54d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang OperandType type1(Type::TENSOR_INT32, {2}); 64d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang // Phase 1, operands 74d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang auto input = model->addOperand(&type0); 84d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang auto block_size = model->addOperand(&type1); 9a9ce0001ca1df32fde3f322c385eef8ce7279329Miao Wang auto output = model->addOperand(&type2); 104d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang // Phase 2, operations 114d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang static int32_t block_size_init[] = {2, 2}; 124d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang model->setOperandValue(block_size, block_size_init, sizeof(int32_t) * 2); 13a9ce0001ca1df32fde3f322c385eef8ce7279329Miao Wang model->addOperation(ANEURALNETWORKS_BATCH_TO_SPACE_ND, {input, block_size}, {output}); 144d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang // Phase 3, inputs and outputs 154d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang model->identifyInputsAndOutputs( 164d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang {input}, 174d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang {output}); 184d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang assert(model->isValid()); 194d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang} 204d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang 214d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wangbool is_ignored(int i) { 224d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang static std::set<int> ignore = {}; 234d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang return ignore.find(i) != ignore.end(); 244d8c48b3cc55c0748ef87a0ba8bbbd3959680213Miao Wang} 25