Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestTrivialModel.cpp253 Model modelBroadcastAdd2; in TEST_F() local
255 auto activation = modelBroadcastAdd2.addConstantOperand(&scalarType, kNoActivation); in TEST_F()
260 auto a = modelBroadcastAdd2.addOperand(&matrixType); in TEST_F()
261 auto b = modelBroadcastAdd2.addOperand(&matrixType2); in TEST_F()
262 auto c = modelBroadcastAdd2.addOperand(&matrixType); in TEST_F()
263 modelBroadcastAdd2.addOperation(ANEURALNETWORKS_ADD, {a, b, activation}, {c}); in TEST_F()
264 modelBroadcastAdd2.identifyInputsAndOutputs({a, b}, {c}); in TEST_F()
265 ASSERT_TRUE(modelBroadcastAdd2.isValid()); in TEST_F()
266 modelBroadcastAdd2.finish(); in TEST_F()
271 Compilation compilation(&modelBroadcastAdd2); in TEST_F()