Searched refs:tensorType (Results 1 – 4 of 4) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestValidation.cpp | 175 ANeuralNetworksOperandType tensorType{.type = ANEURALNETWORKS_TENSOR_FLOAT32, in SetUp() local
180 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &tensorType), ANEURALNETWORKS_NO_ERROR); in SetUp()
729 ANeuralNetworksOperandType tensorType = { in TEST_F() local
953 ANeuralNetworksOperandType tensorType{ in TEST_F() local
983 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &tensorType), ANEURALNETWORKS_NO_ERROR); in TEST_F()
984 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &tensorType), ANEURALNETWORKS_NO_ERROR); in TEST_F()
986 ASSERT_EQ(ANeuralNetworksModel_addOperand(mModel, &tensorType), ANEURALNETWORKS_NO_ERROR); in TEST_F()
1000 ANeuralNetworksOperandType tensorType{ in TEST_F() local
1020 ANeuralNetworksOperandType tensorType{ in TEST_F() local
1041 ANeuralNetworksOperandType tensorType{ in TEST_F() local
[all …]
|
| H A D | TestGpuNnapi.cpp | 838 const OperandType tensorType(dataType, {kOperandSizeY, kOperandSizeX}, scale, in initialize() local
840 uint32_t inputTensor = mModel.addOperand(&tensorType); in initialize()
853 uint32_t outputTensor = mModel.addOperand(&tensorType); in initialize()
|
| H A D | TestExecution.cpp | 726 static const WrapperOperandType tensorType(WrapperType::TENSOR_FLOAT32, {1}); in makeModel() local
729 uint32_t input = model.addOperand(&tensorType); in makeModel()
730 uint32_t output = model.addOperand(&tensorType); in makeModel()
|
| /aosp12/packages/modules/NeuralNetworks/tools/test_generator/ |
| H A D | test_generator.py | 755 tensorType = "TENSOR_QUANT8_ASYMM"
757 tensorType = "TENSOR_QUANT8_ASYMM_SIGNED"
760 tensorType = [tensorType, self.scale, self.zeroPoint]
763 tensorType = ["TENSOR_" + self.targetType.upper()]
770 targets.update({op: tensorType
|