Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestTrivialModel.cpp190 Model modelBroadcastMul2; in TEST_F() local
194 auto activation = modelBroadcastMul2.addOperand(&scalarType); in TEST_F()
195 modelBroadcastMul2.setOperandValue(activation, activation_init, sizeof(int32_t) * 1); in TEST_F()
200 auto a = modelBroadcastMul2.addOperand(&matrixType); in TEST_F()
201 auto b = modelBroadcastMul2.addOperand(&matrixType2); in TEST_F()
202 auto c = modelBroadcastMul2.addOperand(&matrixType); in TEST_F()
203 modelBroadcastMul2.addOperation(ANEURALNETWORKS_MUL, {a, b, activation}, {c}); in TEST_F()
204 modelBroadcastMul2.identifyInputsAndOutputs({a, b}, {c}); in TEST_F()
205 ASSERT_TRUE(modelBroadcastMul2.isValid()); in TEST_F()
206 modelBroadcastMul2.finish(); in TEST_F()
[all …]