Searched refs:scalarType (Results 1 – 8 of 8) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | ResizeImageOps.cpp | 114 auto scalarType = context->getInputType(kOutputHeightParamScalar); in validate() local
115 std::vector<OperandType> inExpectedTypes = {inputType, scalarType, scalarType}; in validate()
123 if (scalarType != OperandType::INT32) { in validate()
126 NN_RET_CHECK(scalarType == OperandType::FLOAT32); in validate()
128 NN_RET_CHECK(scalarType == OperandType::FLOAT16); in validate()
130 NN_RET_CHECK(scalarType == OperandType::FLOAT32); in validate()
161 auto scalarType = context->getInputType(kOutputHeightParamScalar); in prepare() local
162 if (scalarType == OperandType::INT32) { in prepare()
165 } else if (scalarType == OperandType::FLOAT32) { in prepare()
170 } else if (scalarType == OperandType::FLOAT16) { in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestTrivialModel.cpp | 62 OperandType scalarType(Type::INT32, {}); in CreateAddTwoTensorModel() local
67 auto d = model->addOperand(&scalarType); in CreateAddTwoTensorModel()
79 OperandType scalarType(Type::INT32, {}); in CreateAddThreeTensorModel() local
86 auto f = model->addOperand(&scalarType); in CreateAddThreeTensorModel()
161 OperandType scalarType(Type::INT32, {}); in TEST_F() local
162 auto activation = modelBroadcastAdd2.addOperand(&scalarType); in TEST_F()
193 OperandType scalarType(Type::INT32, {}); in TEST_F() local
194 auto activation = modelBroadcastMul2.addOperand(&scalarType); in TEST_F()
|
| D | TestMemory.cpp | 61 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F() local
68 auto f = model.addOperand(&scalarType); in TEST_F()
119 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F() local
126 auto f = model.addOperand(&scalarType); in TEST_F()
|
| D | TestMemoryInternal.cpp | 119 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F() local
126 auto f = model.addOperand(&scalarType); in TEST_F()
|
| D | TestUnknownDimensions.cpp | 199 OperandType scalarType(Type::INT32, {}); in TestOne() local
201 auto activationOpd0 = model.addOperand(&scalarType); in TestOne()
|
| D | TestCompilationCaching.cpp | 224 test_wrapper::OperandType scalarType(Type::INT32, {}); in CreateBroadcastAddModel() local
229 auto d = model->addOperand(&scalarType); in CreateBroadcastAddModel()
|
| D | TestValidation.cpp | 152 ANeuralNetworksOperandType scalarType{ in SetUp() local
156 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &scalarType), ANEURALNETWORKS_NO_ERROR); in SetUp()
|
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciExtensionTest.cpp | 134 ExtensionOperandType scalarType(Type::INT32, {}); in addNopOperation() local
135 uint32_t activation = model->addOperand(&scalarType); in addNopOperation()
|