Searched refs:OperandTypeParams (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | QuantizedLSTMTest.cpp | 32 struct OperandTypeParams { struct
38 OperandTypeParams(Type type, std::vector<uint32_t> shape, float scale, int32_t zeroPoint) in OperandTypeParams() argument
51 QuantizedLSTMOpModel(const std::vector<OperandTypeParams>& inputOperandTypeParams) { in QuantizedLSTMOpModel()
186 void initializeInputData(OperandTypeParams params, std::vector<T>* vec) { in initializeInputData()
292 OperandTypeParams(Type::TENSOR_QUANT8_ASYMM, {outputSize, inputSize}, weightsScale, in TEST_F()
294 OperandTypeParams(Type::TENSOR_QUANT8_ASYMM, {outputSize, inputSize}, weightsScale, in TEST_F()
296 OperandTypeParams(Type::TENSOR_QUANT8_ASYMM, {outputSize, inputSize}, weightsScale, in TEST_F()
316 OperandTypeParams(Type::TENSOR_INT32, {outputSize}, weightsScale / 128., 0), in TEST_F()
317 OperandTypeParams(Type::TENSOR_INT32, {outputSize}, weightsScale / 128., 0), in TEST_F()
318 OperandTypeParams(Type::TENSOR_INT32, {outputSize}, weightsScale / 128., 0), in TEST_F()
[all …]
|