Searched refs:kForgetLayerNormTensor (Results 1 – 1 of 1) sorted by relevance
77 constexpr uint32_t kForgetLayerNormTensor = 21; variable313 if (hasTensor(context, kForgetLayerNormTensor)) { in prepare()314 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in prepare()334 const bool layerNormWeightsAllOrNoneCifg = (hasTensor(context, kForgetLayerNormTensor) && in prepare()337 (!hasTensor(context, kForgetLayerNormTensor) && in prepare()343 hasTensor(context, kForgetLayerNormTensor) && in prepare()347 !hasTensor(context, kForgetLayerNormTensor) && in prepare()379 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in execute()428 reinterpret_cast<const int16_t*>(context->getInputBuffer(kForgetLayerNormTensor)); in execute()