Home
last modified time | relevance | path

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

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQLSTM.cpp209 if (hasTensor(context, kOutputLayerNormTensor)) { in prepare()
210 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in prepare()
220 hasTensor(context, kOutputLayerNormTensor)) || in prepare()
223 !hasTensor(context, kOutputLayerNormTensor)); in prepare()
229 hasTensor(context, kOutputLayerNormTensor)) || in prepare()
233 !hasTensor(context, kOutputLayerNormTensor)); in prepare()
271 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormTensor); in execute()
332 reinterpret_cast<const int16_t*>(context->getInputBuffer(kOutputLayerNormTensor)); in execute()
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DQLSTM.h69 constexpr uint32_t kOutputLayerNormTensor = 23; variable