Searched refs:kInputLayerNormTensor (Results 1 – 1 of 1) sorted by relevance
76 constexpr uint32_t kInputLayerNormTensor = 20; variable307 if (hasTensor(context, kInputLayerNormTensor)) { in prepare()308 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in prepare()332 NN_RET_CHECK(!hasTensor(context, kInputLayerNormTensor)) in prepare()342 const bool layerNormWeightsAllOrNone = (hasTensor(context, kInputLayerNormTensor) && in prepare()346 (!hasTensor(context, kInputLayerNormTensor) && in prepare()374 const Shape inputLayerNormShape = context->getInputShape(kInputLayerNormTensor); in execute()417 reinterpret_cast<const int16_t*>(context->getInputBuffer(kInputLayerNormTensor)); in execute()