Searched refs:outputLayerNormShape (Results 1 – 2 of 2) sorted by relevance
325 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in prepare() local326 NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1); in prepare()327 NN_RET_CHECK_EQ(getSizeOfDimension(outputLayerNormShape, 0), numUnits); in prepare()385 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in execute() local575 NN_RET_CHECK(QuantizeMultiplier(outputLayerNormShape.scale, &outputLayerNormScaleA, in execute()669 std::min(1, static_cast<int32_t>(10000 * outputLayerNormShape.scale)); in execute()
345 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormWeightsTensor); in prepare() local346 NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1); in prepare()347 NN_RET_CHECK_EQ(getSizeOfDimension(outputLayerNormShape, 0), numCells); in prepare()