1 // clang-format off
2 // Generated file (from: concat_float_1.mod.py). Do not edit
CreateModel(Model * model)3 void CreateModel(Model *model) {
4 OperandType type0(Type::TENSOR_FLOAT32, {2, 3});
5 OperandType type1(Type::INT32, {});
6 OperandType type2(Type::TENSOR_FLOAT32, {4, 3});
7 // Phase 1, operands
8 auto op1 = model->addOperand(&type0);
9 auto op2 = model->addOperand(&type0);
10 auto axis0 = model->addOperand(&type1);
11 auto result = model->addOperand(&type2);
12 // Phase 2, operations
13 static int32_t axis0_init[] = {0};
14 model->setOperandValue(axis0, axis0_init, sizeof(int32_t) * 1);
15 model->addOperation(ANEURALNETWORKS_CONCATENATION, {op1, op2, axis0}, {result});
16 // Phase 3, inputs and outputs
17 model->identifyInputsAndOutputs(
18 {op1, op2},
19 {result});
20 assert(model->isValid());
21 }
22
is_ignored(int i)23 inline bool is_ignored(int i) {
24 static std::set<int> ignore = {};
25 return ignore.find(i) != ignore.end();
26 }
27
CreateModel_dynamic_output_shape(Model * model)28 void CreateModel_dynamic_output_shape(Model *model) {
29 OperandType type0(Type::TENSOR_FLOAT32, {2, 3});
30 OperandType type1(Type::INT32, {});
31 OperandType type3(Type::TENSOR_FLOAT32, {0, 0});
32 // Phase 1, operands
33 auto op1 = model->addOperand(&type0);
34 auto op2 = model->addOperand(&type0);
35 auto axis0 = model->addOperand(&type1);
36 auto result = model->addOperand(&type3);
37 // Phase 2, operations
38 static int32_t axis0_init[] = {0};
39 model->setOperandValue(axis0, axis0_init, sizeof(int32_t) * 1);
40 model->addOperation(ANEURALNETWORKS_CONCATENATION, {op1, op2, axis0}, {result});
41 // Phase 3, inputs and outputs
42 model->identifyInputsAndOutputs(
43 {op1, op2},
44 {result});
45 assert(model->isValid());
46 }
47
is_ignored_dynamic_output_shape(int i)48 inline bool is_ignored_dynamic_output_shape(int i) {
49 static std::set<int> ignore = {};
50 return ignore.find(i) != ignore.end();
51 }
52
53