Lines Matching refs:numBatches
59 const uint32_t numBatches = inputOperandTypeParams[0].shape[0]; in QuantizedLSTMOpModel() local
66 OperandType cellStateOutOperandType(Type::TENSOR_QUANT16_SYMM, {numBatches, outputSize}, in QuantizedLSTMOpModel()
69 OperandType outputOperandType(Type::TENSOR_QUANT8_ASYMM, {numBatches, outputSize}, in QuantizedLSTMOpModel()
82 cellStateOut_.resize(numBatches * outputSize, 0); in QuantizedLSTMOpModel()
83 output_.resize(numBatches * outputSize, 0); in QuantizedLSTMOpModel()
239 const int numBatches = input.size(); in VerifyGoldens() local
240 EXPECT_GT(numBatches, 0); in VerifyGoldens()
247 for (int b = 0; b < numBatches; ++b) { in VerifyGoldens()
257 for (int b = 0; b < numBatches; ++b) { in VerifyGoldens()
271 const int numBatches = 2; in TEST_F() local
286 OperandTypeParams(Type::TENSOR_QUANT8_ASYMM, {numBatches, inputSize}, 1. / 128., 128), in TEST_F()
320 OperandTypeParams(Type::TENSOR_QUANT16_SYMM, {numBatches, outputSize}, 1. / 2048., 0), in TEST_F()
322 OperandTypeParams(Type::TENSOR_QUANT8_ASYMM, {numBatches, outputSize}, 1. / 128., 128), in TEST_F()