Searched refs:inputLayerNormShape (Results 1 – 2 of 2) sorted by relevance
308 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in prepare() local309 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1); in prepare()310 NN_RET_CHECK_EQ(getSizeOfDimension(inputLayerNormShape, 0), numUnits); in prepare()374 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in execute() local501 NN_RET_CHECK(QuantizeMultiplier(inputLayerNormShape.scale, &inputLayerNormScaleA, in execute()665 std::min(1, static_cast<int32_t>(10000 * inputLayerNormShape.scale)); in execute()
330 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormWeightsTensor); in prepare() local331 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1); in prepare()332 NN_RET_CHECK_EQ(getSizeOfDimension(inputLayerNormShape, 0), numCells); in prepare()