Home
last modified time | relevance | path

Searched refs:kCellStateInTensor (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp79 kCellStateInTensor, in prepare()
215 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare()
281 Shape cellStateOutTensor = context->getInputShape(kCellStateInTensor); in prepare()
293 const auto cellStateSize = getNumberOfElements(context->getInputShape(kCellStateInTensor)); in execute()
344 context->getInputBuffer<float>(kCellStateInTensor), in execute()
397 context->getInputBuffer<_Float16>(kCellStateInTensor), in execute()
DLSTM.cpp84 cell_state_in_ = GetInput(operation, operands, kCellStateInTensor); in LSTMCell()
321 kCellStateInTensor, in Prepare()
/packages/modules/NeuralNetworks/common/types/operations/include/
DUnidirectionalSequenceLSTM.h61 constexpr uint32_t kCellStateInTensor = 19; variable
DLSTM.h88 static constexpr int kCellStateInTensor = 19; variable