Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestTrivialModel.cpp282 Model modelBroadcastMul2; in TEST_F() local
284 auto activation = modelBroadcastMul2.addConstantOperand(&scalarType, kNoActivation); in TEST_F()
289 auto a = modelBroadcastMul2.addOperand(&matrixType); in TEST_F()
290 auto b = modelBroadcastMul2.addOperand(&matrixType2); in TEST_F()
291 auto c = modelBroadcastMul2.addOperand(&matrixType); in TEST_F()
292 modelBroadcastMul2.addOperation(ANEURALNETWORKS_MUL, {a, b, activation}, {c}); in TEST_F()
293 modelBroadcastMul2.identifyInputsAndOutputs({a, b}, {c}); in TEST_F()
294 ASSERT_TRUE(modelBroadcastMul2.isValid()); in TEST_F()
295 modelBroadcastMul2.finish(); in TEST_F()
300 Compilation compilation(&modelBroadcastMul2); in TEST_F()