Searched defs:modelBroadcastMul2 (Results 1 - 1 of 1) sorted by relevance

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

Completed in 57 milliseconds