Home
last modified time | relevance | path

Searched refs:kOutputLayerNormTensor (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQLSTM.cpp79 constexpr uint32_t kOutputLayerNormTensor = 23; variable
325 if (hasTensor(context, kOutputLayerNormTensor)) { in prepare()
326 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in prepare()
336 hasTensor(context, kOutputLayerNormTensor)) || in prepare()
339 !hasTensor(context, kOutputLayerNormTensor)); in prepare()
345 hasTensor(context, kOutputLayerNormTensor)) || in prepare()
349 !hasTensor(context, kOutputLayerNormTensor)); in prepare()
387 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in execute()
448 reinterpret_cast<const int16_t*>(context->getInputBuffer(kOutputLayerNormTensor)); in execute()