Searched refs:conditionModel (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestControlFlow.cpp | 48 Model conditionModel; in TEST_F() local
50 uint32_t i = conditionModel.addOperand(&counterType); in TEST_F()
51 uint32_t n = conditionModel.addOperand(&counterType); in TEST_F()
52 uint32_t out = conditionModel.addOperand(&boolType); in TEST_F()
53 conditionModel.addOperation(ANEURALNETWORKS_GREATER_EQUAL, {i, n}, {out}); in TEST_F()
54 conditionModel.identifyInputsAndOutputs({i, n}, {out}); in TEST_F()
55 ASSERT_EQ(conditionModel.finish(), Result::NO_ERROR); in TEST_F()
56 ASSERT_TRUE(conditionModel.isValid()); in TEST_F()
76 uint32_t conditionOperand = model.addModelOperand(&conditionModel); in TEST_F()
|
| D | TestFailingDriver.cpp | 108 WrapperModel conditionModel; in TEST_F() local
110 uint32_t f = conditionModel.addOperand(&floatType); in TEST_F()
111 uint32_t b = conditionModel.addOperand(&boolType); in TEST_F()
112 uint32_t out = conditionModel.addOperand(&boolType); in TEST_F()
113 conditionModel.addOperation(ANEURALNETWORKS_CAST, {b}, {out}); in TEST_F()
114 conditionModel.identifyInputsAndOutputs({f, b}, {out}); in TEST_F()
115 ASSERT_EQ(conditionModel.finish(), Result::NO_ERROR); in TEST_F()
116 ASSERT_TRUE(conditionModel.isValid()); in TEST_F()
136 uint32_t cond = model.addModelOperand(&conditionModel); in TEST_F()
|