Home
last modified time | relevance | path

Searched refs:modelBroadcastAdd2 (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/runtime/test/
DTestTrivialModel.cpp158 Model modelBroadcastAdd2; in TEST_F() local
162 auto activation = modelBroadcastAdd2.addOperand(&scalarType); in TEST_F()
163 modelBroadcastAdd2.setOperandValue(activation, activation_init, sizeof(int32_t) * 1); in TEST_F()
168 auto a = modelBroadcastAdd2.addOperand(&matrixType); in TEST_F()
169 auto b = modelBroadcastAdd2.addOperand(&matrixType2); in TEST_F()
170 auto c = modelBroadcastAdd2.addOperand(&matrixType); in TEST_F()
171 modelBroadcastAdd2.addOperation(ANEURALNETWORKS_ADD, {a, b, activation}, {c}); in TEST_F()
172 modelBroadcastAdd2.identifyInputsAndOutputs({a, b}, {c}); in TEST_F()
173 ASSERT_TRUE(modelBroadcastAdd2.isValid()); in TEST_F()
174 modelBroadcastAdd2.finish(); in TEST_F()
[all …]