Searched refs:inputLayerNormShape (Results 1 – 2 of 2) sorted by relevance
307 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in prepare() local308 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1); in prepare()309 NN_RET_CHECK_EQ(getSizeOfDimension(inputLayerNormShape, 0), numUnits); in prepare()372 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in execute() local499 NN_RET_CHECK(QuantizeMultiplier(inputLayerNormShape.scale, &inputLayerNormScaleA, in execute()663 std::min(1, static_cast<int32_t>(10000 * inputLayerNormShape.scale)); in execute()
327 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormWeightsTensor); in prepare() local328 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1); in prepare()329 NN_RET_CHECK_EQ(getSizeOfDimension(inputLayerNormShape, 0), numCells); in prepare()