Searched refs:kOutputLayerNormWeightsTensor (Results 1 – 3 of 3) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceLSTM.cpp | 81 constexpr uint32_t kOutputLayerNormWeightsTensor = 27; // Optional variable
111 params.use_layer_norm = hasTensor(context, kOutputLayerNormWeightsTensor); in getLSTMParams()
347 if (hasTensor(context, kOutputLayerNormWeightsTensor)) { in prepare()
348 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormWeightsTensor); in prepare()
359 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
362 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
369 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
373 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
457 context->getInputBuffer<float>(kOutputLayerNormWeightsTensor), outputStateOut, in execute()
510 context->getInputBuffer<_Float16>(kOutputLayerNormWeightsTensor), in execute()
|
| H A D | LSTM.h | 99 static constexpr int kOutputLayerNormWeightsTensor = 26; variable
|
| H A D | LSTM.cpp | 112 GetInput(operation, operands, kOutputLayerNormWeightsTensor); // optional in LSTMCell()
|