Lines Matching refs:bias
301 auto checkBiasShape = [&](const RunTimeOperandInfo* bias) -> bool { in prepare() argument
302 NN_RET_CHECK_EQ(NumDimensions(bias), 1); in prepare()
303 NN_RET_CHECK_EQ(SizeOfDimension(bias, 0), outputSize); in prepare()
304 NN_RET_CHECK_EQ(bias->scale, biasScale); in prepare()
305 NN_RET_CHECK_EQ(bias->zeroPoint, biasZeroPoint); in prepare()
368 void QuantizedLSTMCell::concatenateBiases(uint32_t outputSize, int32_t* bias) { in concatenateBiases() argument
369 memcpy(bias + 0 * outputSize, GetBuffer<int32_t>(inputGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
370 memcpy(bias + 1 * outputSize, GetBuffer<int32_t>(cellGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
371 memcpy(bias + 2 * outputSize, GetBuffer<int32_t>(forgetGateBias_), in concatenateBiases()
373 memcpy(bias + 3 * outputSize, GetBuffer<int32_t>(outputGateBias_), in concatenateBiases()
388 std::vector<int32_t> bias(getNumberOfElements(biasShape)); in eval() local
389 concatenateBiases(SizeOfDimension(prevOutput_, 1), bias.data()); in eval()
417 weights.data(), convertShapeToDims(weightsShape), bias.data(), in eval()