Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestTrivialModel.cpp273 Model modelBroadcastAdd2; in TEST_F() local
275 auto activation = modelBroadcastAdd2.addConstantOperand(&scalarType, kNoActivation); in TEST_F()
280 auto a = modelBroadcastAdd2.addOperand(&matrixType); in TEST_F()
281 auto b = modelBroadcastAdd2.addOperand(&matrixType2); in TEST_F()
282 auto c = modelBroadcastAdd2.addOperand(&matrixType); in TEST_F()
283 modelBroadcastAdd2.addOperation(ANEURALNETWORKS_ADD, {a, b, activation}, {c}); in TEST_F()
284 modelBroadcastAdd2.identifyInputsAndOutputs({a, b}, {c}); in TEST_F()
285 ASSERT_TRUE(modelBroadcastAdd2.isValid()); in TEST_F()
286 modelBroadcastAdd2.finish(); in TEST_F()
291 Compilation compilation(&modelBroadcastAdd2); in TEST_F()