Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQLSTM.cpp78 constexpr uint32_t kCellLayerNormTensor = 22; variable
319 if (hasTensor(context, kCellLayerNormTensor)) { in prepare()
320 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in prepare()
335 hasTensor(context, kCellLayerNormTensor) && in prepare()
338 !hasTensor(context, kCellLayerNormTensor) && in prepare()
344 hasTensor(context, kCellLayerNormTensor) && in prepare()
348 !hasTensor(context, kCellLayerNormTensor) && in prepare()
382 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormTensor); in execute()
437 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellLayerNormTensor)); in execute()