Lines Matching refs:bias
298 auto checkBiasShape = [&](const RunTimeOperandInfo* bias) -> bool { in prepare() argument
299 NN_RET_CHECK_EQ(NumDimensions(bias), 1); in prepare()
300 NN_RET_CHECK_EQ(SizeOfDimension(bias, 0), outputSize); in prepare()
301 NN_RET_CHECK_EQ(bias->scale, biasScale); in prepare()
302 NN_RET_CHECK_EQ(bias->zeroPoint, biasZeroPoint); in prepare()
365 void QuantizedLSTMCell::concatenateBiases(uint32_t outputSize, int32_t* bias) { in concatenateBiases() argument
366 memcpy(bias + 0 * outputSize, GetBuffer<int32_t>(inputGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
367 memcpy(bias + 1 * outputSize, GetBuffer<int32_t>(cellGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
368 memcpy(bias + 2 * outputSize, GetBuffer<int32_t>(forgetGateBias_), in concatenateBiases()
370 memcpy(bias + 3 * outputSize, GetBuffer<int32_t>(outputGateBias_), in concatenateBiases()
385 std::vector<int32_t> bias(getNumberOfElements(biasShape)); in eval() local
386 concatenateBiases(SizeOfDimension(prevOutput_, 1), bias.data()); in eval()
414 weights.data(), convertShapeToDims(weightsShape), bias.data(), in eval()