Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DResizeImageOps.cpp185 auto scalarType = context->getInputType(kOutputHeightParamScalar); in validate() local
186 std::vector<OperandType> inExpectedTypes = {inputType, scalarType, scalarType}; in validate()
198 if (scalarType != OperandType::INT32) { in validate()
201 NN_RET_CHECK(scalarType == OperandType::FLOAT32); in validate()
203 NN_RET_CHECK(scalarType == OperandType::FLOAT16); in validate()
206 NN_RET_CHECK(scalarType == OperandType::FLOAT32); in validate()
244 auto scalarType = context->getInputType(kOutputHeightParamScalar); in prepare() local
245 if (scalarType == OperandType::INT32) { in prepare()
248 } else if (scalarType == OperandType::FLOAT32) { in prepare()
253 } else if (scalarType == OperandType::FLOAT16) { in prepare()
DElu.cpp67 auto scalarType = in validate() local
69 return validateInputTypes(context, {inputType, scalarType}) && in validate()
/frameworks/ml/nn/runtime/test/
DTestTrivialModel.cpp62 OperandType scalarType(Type::INT32, {}); in CreateAddTwoTensorModel() local
66 auto d = model->addConstantOperand(&scalarType, kNoActivation); in CreateAddTwoTensorModel()
77 OperandType scalarType(Type::INT32, {}); in CreateAddThreeTensorModel() local
83 auto f = model->addConstantOperand(&scalarType, kNoActivation); in CreateAddThreeTensorModel()
254 OperandType scalarType(Type::INT32, {}); in TEST_F() local
255 auto activation = modelBroadcastAdd2.addConstantOperand(&scalarType, kNoActivation); in TEST_F()
283 OperandType scalarType(Type::INT32, {}); in TEST_F() local
284 auto activation = modelBroadcastMul2.addConstantOperand(&scalarType, kNoActivation); in TEST_F()
DTestMemory.cpp60 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F() local
67 auto f = model.addOperand(&scalarType); in TEST_F()
118 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F() local
125 auto f = model.addOperand(&scalarType); in TEST_F()
DTestMemoryInternal.cpp119 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F() local
126 auto f = model.addOperand(&scalarType); in TEST_F()
DTestUnknownDimensions.cpp216 OperandType scalarType(Type::INT32, {}); in TestOne() local
218 auto activationOpd0 = model.addOperand(&scalarType); in TestOne()
DTestValidation.cpp174 ANeuralNetworksOperandType scalarType{ in SetUp() local
178 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &scalarType), ANEURALNETWORKS_NO_ERROR); in SetUp()
728 ANeuralNetworksOperandType scalarType = {.type = ANEURALNETWORKS_INT32}; in TEST_F() local
735 ASSERT_EQ(ANeuralNetworksModel_addOperand(valueModel, &scalarType), ANEURALNETWORKS_NO_ERROR); in TEST_F()
949 ANeuralNetworksOperandType scalarType{ in TEST_F() local
979 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &scalarType), ANEURALNETWORKS_NO_ERROR); in TEST_F()
1055 ANeuralNetworksOperandType scalarType{ in TEST_F() local
1062 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &scalarType), ANEURALNETWORKS_NO_ERROR); in TEST_F()
3094 const ANeuralNetworksOperandType scalarType = { in createAndCompileChannelQuantConvModel() local
3102 EXPECT_EQ(ANeuralNetworksModel_addOperand(model, &scalarType), ANEURALNETWORKS_NO_ERROR); in createAndCompileChannelQuantConvModel()
[all …]
DTestCompilationCaching.cpp272 test_wrapper::OperandType scalarType(Type::INT32, {}); in CreateBroadcastAddModel() local
277 auto d = model->addOperand(&scalarType); in CreateBroadcastAddModel()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciExtensionTest.cpp142 ExtensionOperandType scalarType(Type::INT32, {}); in addNopOperation() local
143 uint32_t activation = model->addOperand(&scalarType); in addNopOperation()
/frameworks/ml/nn/tools/test_generator/
Dtest_generator.py761 scalarType = None # Not supported.
764 scalarType = [self.targetType.upper()]
772 if scalarType is not None:
773 targets.update({op: scalarType