Searched refs:kOutputLayerNormWeightsTensor (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 77 constexpr uint32_t kOutputLayerNormWeightsTensor = 27; // Optional variable
103 params.use_layer_norm = hasTensor(context, kOutputLayerNormWeightsTensor); in getLSTMParams()
331 if (hasTensor(context, kOutputLayerNormWeightsTensor)) { in prepare()
332 const Shape outputLayerNormShape = context->getInputShape(kOutputLayerNormWeightsTensor); in prepare()
343 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
346 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
353 hasTensor(context, kOutputLayerNormWeightsTensor)) || in prepare()
357 !hasTensor(context, kOutputLayerNormWeightsTensor)); in prepare()
411 context->getInputBuffer<float>(kOutputLayerNormWeightsTensor), in execute()
453 context->getInputBuffer<_Float16>(kOutputLayerNormWeightsTensor), in execute()
|
| D | LSTM.h | 97 static constexpr int kOutputLayerNormWeightsTensor = 26; variable
|
| D | LSTM.cpp | 103 GetInput(operation, operands, kOutputLayerNormWeightsTensor); // optional in LSTMCell()
|