Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp76 constexpr uint32_t kCellLayerNormWeightsTensor = 26; // Optional variable
325 if (hasTensor(context, kCellLayerNormWeightsTensor)) { in prepare()
326 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormWeightsTensor); in prepare()
342 hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
345 !hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
352 hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
356 !hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
410 context->getInputBuffer<float>(kCellLayerNormWeightsTensor), in execute()
452 context->getInputBuffer<_Float16>(kCellLayerNormWeightsTensor), in execute()
DLSTM.h96 static constexpr int kCellLayerNormWeightsTensor = 25; variable
DLSTM.cpp101 GetInput(operation, operands, kCellLayerNormWeightsTensor); // optional in LSTMCell()