Home
last modified time | relevance | path

Searched refs:kCellToInputWeightsTensor (Results 1 – 8 of 8) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp149 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
150 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
170 ((hasTensor(context, kCellToInputWeightsTensor) || cifgUsed) && in prepare()
173 (!hasTensor(context, kCellToInputWeightsTensor) && in prepare()
329 context->getInputBuffer<float>(kCellToInputWeightsTensor), in execute()
382 context->getInputBuffer<_Float16>(kCellToInputWeightsTensor), in execute()
DQLSTM.cpp120 if (hasTensor(context, kCellToInputWeightsTensor)) { in prepare()
121 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare()
141 ((hasTensor(context, kCellToInputWeightsTensor) || cifgUsed) && in prepare()
144 (!hasTensor(context, kCellToInputWeightsTensor) && in prepare()
257 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute()
299 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToInputWeightsTensor)); in execute()
DLayerNormLSTMTest.cpp235 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
238 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
DLSTMTest.cpp222 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
225 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
DLSTM.cpp69 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
/packages/modules/NeuralNetworks/common/types/operations/include/
DUnidirectionalSequenceLSTM.h43 constexpr uint32_t kCellToInputWeightsTensor = 9; // Optional variable
DQLSTM.h42 constexpr uint32_t kCellToInputWeightsTensor = 9; variable
DLSTM.h72 static constexpr int kCellToInputWeightsTensor = 9; // Optional variable