Searched refs:outputOperandType (Results 1 – 2 of 2) sorted by relevance
/aosp12/packages/modules/NeuralNetworks/common/operations/ |
H A D | QuantizedLSTMTest.cpp | 70 OperandType outputOperandType(Type::TENSOR_QUANT8_ASYMM, {numBatches, outputSize}, in QuantizedLSTMOpModel() local 72 outputs.push_back(model_.addOperand(&outputOperandType)); in QuantizedLSTMOpModel()
|
/aosp12/packages/modules/NeuralNetworks/runtime/test/ |
H A D | TestValidateOperations.cpp | 1063 void dequantizeOpTest(int32_t inputOperandType, int32_t outputOperandType) { in dequantizeOpTest() argument 1065 << "inputType: " << inputOperandType << ", outputType: " << outputOperandType); in dequantizeOpTest() 1068 ANeuralNetworksOperandType output = getOpType(outputOperandType, 4, inputDimensions); in dequantizeOpTest() 4415 void fillTest(int32_t valueOperandType, int32_t outputOperandType) { in fillTest() argument 4420 ANeuralNetworksOperandType output = getOpType(outputOperandType, 3, outputDimensions); in fillTest()
|