Searched refs:kOutputLayerNormWeightsTensor (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 57 params.use_layer_norm = hasTensor(context, kOutputLayerNormWeightsTensor); in getLSTMParams()
238 if (hasTensor(context, kOutputLayerNormWeightsTensor)) { in prepare()
239 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormWeightsTensor); in prepare()
250 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
253 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
260 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
264 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
348 context->getInputBuffer<float>(kOutputLayerNormWeightsTensor), outputStateOut, in execute()
401 context->getInputBuffer<_Float16>(kOutputLayerNormWeightsTensor), in execute()
|
| D | LSTM.cpp | 112 GetInput(operation, operands, kOutputLayerNormWeightsTensor); // optional in LSTMCell()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 72 constexpr uint32_t kOutputLayerNormWeightsTensor = 27; // Optional variable
|
| D | LSTM.h | 98 static constexpr int kOutputLayerNormWeightsTensor = 26; variable
|