Searched refs:LSTMCell (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LayerNormLSTMTest.cpp | 210 execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
220 execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
228 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
229 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
234 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
237 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
238 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
239 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
244 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
247 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
[all …]
|
| D | LSTMTest.cpp | 200 execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
209 execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), sizeof(float) * X##_.size()), \ in Invoke()
217 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
218 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
223 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
226 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
227 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
228 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
233 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
236 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
[all …]
|
| D | LSTM.h | 49 class LSTMCell {
51 LSTMCell(const hal::Operation& operation, RunTimeOperandInfo* operands);
|
| D | BidirectionalSequenceLSTM.cpp | 288 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
396 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
485 LSTMCell::LSTMEvalFloat32( in Eval()
537 LSTMCell::LSTMEvalFloat32( in Eval()
609 LSTMCell::LSTMEvalFloat16( in Eval()
662 LSTMCell::LSTMEvalFloat16( in Eval()
|
| D | LSTM.cpp | 54 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()
994 bool LSTMCell::Eval() { in Eval()
|
| D | UnidirectionalSequenceLSTM.cpp | 422 LSTMCell::LSTMEvalFloat32( in execute()
474 LSTMCell::LSTMEvalFloat16( in execute()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1095 RunTimeOperandInfo& scratch = operands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1096 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1097 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1098 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1101 LSTMCell lstm_cell(operation, operands); in executeOperation()
|