Searched refs:kOutputLayerNormWeightsTensor (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 79 constexpr uint32_t kOutputLayerNormWeightsTensor = 27; // Optional variable
110 params.use_layer_norm = hasTensor(context, kOutputLayerNormWeightsTensor); in getLSTMParams()
344 if (hasTensor(context, kOutputLayerNormWeightsTensor)) { in prepare()
345 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormWeightsTensor); in prepare()
356 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
359 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
366 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
370 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
454 context->getInputBuffer<float>(kOutputLayerNormWeightsTensor), outputStateOut, in execute()
507 context->getInputBuffer<_Float16>(kOutputLayerNormWeightsTensor), in execute()
|
| D | LSTM.h | 99 static constexpr int kOutputLayerNormWeightsTensor = 26; variable
|
| D | LSTM.cpp | 112 GetInput(operation, operands, kOutputLayerNormWeightsTensor); // optional in LSTMCell()
|