Home
last modified time | relevance | path

Searched refs:LSTMCell (Results 1 – 7 of 7) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DLayerNormLSTMTest.cpp211 execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
221 execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
229 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
230 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
235 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
238 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
239 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
240 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
245 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
248 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
[all …]
DLSTMTest.cpp199 execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
208 execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
216 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
217 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
222 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
225 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
226 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
227 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
232 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
235 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
[all …]
DBidirectionalSequenceLSTM.cpp287 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
395 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
484 LSTMCell::LSTMEvalFloat32( in Eval()
536 LSTMCell::LSTMEvalFloat32( in Eval()
608 LSTMCell::LSTMEvalFloat16( in Eval()
661 LSTMCell::LSTMEvalFloat16( in Eval()
DLSTM.cpp54 LSTMCell::LSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in LSTMCell() function in android::nn::LSTMCell
132 bool LSTMCell::CheckInputTensorDimensions( in CheckInputTensorDimensions()
303 bool LSTMCell::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
405 bool LSTMCell::LSTMEvalFloat32( in LSTMEvalFloat32()
521 bool LSTMCell::LSTMEvalFloat16( in LSTMEvalFloat16()
765 bool LSTMCell::LSTMStep( in LSTMStep()
992 bool LSTMCell::Eval() { in Eval()
DUnidirectionalSequenceLSTM.cpp316 LSTMCell::LSTMEvalFloat32( in execute()
368 LSTMCell::LSTMEvalFloat16( in execute()
/packages/modules/NeuralNetworks/common/types/operations/include/
DLSTM.h48 class LSTMCell {
50 LSTMCell(const Operation& operation, RunTimeOperandInfo* operands);
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1042 RunTimeOperandInfo& scratch = operands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1043 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1044 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1045 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1048 LSTMCell lstm_cell(operation, operands); in executeOperation()