Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp78 constexpr uint32_t kCellLayerNormWeightsTensor = 26; // Optional variable
338 if (hasTensor(context, kCellLayerNormWeightsTensor)) { in prepare()
339 const Shape cellLayerNormShape = context->getInputShape(kCellLayerNormWeightsTensor); in prepare()
355 hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
358 !hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
365 hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
369 !hasTensor(context, kCellLayerNormWeightsTensor) && in prepare()
453 context->getInputBuffer<float>(kCellLayerNormWeightsTensor), in execute()
506 context->getInputBuffer<_Float16>(kCellLayerNormWeightsTensor), in execute()
DLSTM.h98 static constexpr int kCellLayerNormWeightsTensor = 25; variable
DLSTM.cpp110 GetInput(operation, operands, kCellLayerNormWeightsTensor); // optional in LSTMCell()