Lines Matching refs:model
123 WrapperModel model; in TEST_F() local
127 auto a = model.addOperand(&matrixType); in TEST_F()
128 auto b = model.addOperand(&matrixType); in TEST_F()
129 auto c = model.addOperand(&matrixType); in TEST_F()
130 auto d = model.addOperand(&matrixType); in TEST_F()
131 auto e = model.addOperand(&matrixType); in TEST_F()
132 auto f = model.addOperand(&scalarType); in TEST_F()
134 model.setOperandValueFromMemory(e, &weights, offsetForMatrix2, sizeof(Matrix3x4)); in TEST_F()
135 model.setOperandValueFromMemory(a, &weights, offsetForMatrix3, sizeof(Matrix3x4)); in TEST_F()
136 model.setOperandValue(f, &activation, sizeof(activation)); in TEST_F()
137 model.addOperation(ANEURALNETWORKS_ADD, {a, c, f}, {b}); in TEST_F()
138 model.addOperation(ANEURALNETWORKS_ADD, {b, e, f}, {d}); in TEST_F()
139 model.identifyInputsAndOutputs({c}, {d}); in TEST_F()
140 ASSERT_TRUE(model.isValid()); in TEST_F()
141 model.finish(); in TEST_F()
178 WrapperCompilation compilation2(&model); in TEST_F()
203 WrapperModel model; in TEST_F() local
215 auto op1 = model.addOperand(&type0); in TEST_F()
216 auto op2 = model.addOperand(&type1); in TEST_F()
217 auto op3 = model.addOperand(&type2); in TEST_F()
218 auto pad0 = model.addOperand(&type3); in TEST_F()
219 auto act = model.addOperand(&type3); in TEST_F()
220 auto stride = model.addOperand(&type3); in TEST_F()
221 auto op4 = model.addOperand(&type4); in TEST_F()
225 model.setOperandValue(op2, op2_init, sizeof(op2_init)); in TEST_F()
227 model.setOperandValue(op3, op3_init, sizeof(op3_init)); in TEST_F()
229 model.setOperandValue(pad0, pad0_init, sizeof(pad0_init)); in TEST_F()
231 model.setOperandValue(act, act_init, sizeof(act_init)); in TEST_F()
233 model.setOperandValue(stride, stride_init, sizeof(stride_init)); in TEST_F()
234 model.addOperation(ANEURALNETWORKS_CONV_2D, in TEST_F()
238 model.identifyInputsAndOutputs({op1}, {op4}); in TEST_F()
239 ASSERT_TRUE(model.isValid()); in TEST_F()
240 model.finish(); in TEST_F()
243 WrapperCompilation compilation(&model); in TEST_F()