Searched refs:LSTMCell (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LayerNormLSTMTest.cpp | 211 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 …]
|
| D | LSTMTest.cpp | 195 ASSERT_EQ(execution.setInput(LSTMCell::k##X##Tensor, X##_.data(), \ in Invoke()
204 ASSERT_EQ(execution.setOutput(LSTMCell::k##X##Tensor, X##_.data(), \ in Invoke()
213 execution.setInput(LSTMCell::kInputToInputWeightsTensor, nullptr, 0); in Invoke()
214 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
219 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
222 execution.setInput(LSTMCell::kCellToInputWeightsTensor, nullptr, 0); in Invoke()
223 execution.setInput(LSTMCell::kCellToForgetWeightsTensor, nullptr, 0); in Invoke()
224 execution.setInput(LSTMCell::kCellToOutputWeightsTensor, nullptr, 0); in Invoke()
229 execution.setInput(LSTMCell::kProjectionBiasTensor, nullptr, 0); in Invoke()
232 execution.setInput(LSTMCell::kProjectionWeightsTensor, nullptr, 0); in Invoke()
[all …]
|
| D | BidirectionalSequenceLSTM.cpp | 179 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
230 if (!LSTMCell::CheckInputTensorDimensions( in Prepare()
282 LSTMCell::LSTMEvalFloat32( in Eval()
320 LSTMCell::LSTMEvalFloat32( in Eval()
368 LSTMCell::LSTMEvalFloat16( in Eval()
406 LSTMCell::LSTMEvalFloat16( in Eval()
|
| D | LSTM.h | 46 class LSTMCell {
48 LSTMCell(const Operation& operation, std::vector<RunTimeOperandInfo>& operands);
|
| D | LSTM.cpp | 49 LSTMCell::LSTMCell(const Operation& operation, std::vector<RunTimeOperandInfo>& operands) { in LSTMCell() function in android::nn::LSTMCell
123 bool LSTMCell::CheckInputTensorDimensions( in CheckInputTensorDimensions()
294 bool LSTMCell::Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands, in Prepare()
362 bool LSTMCell::LSTMEvalFloat32( in LSTMEvalFloat32()
478 bool LSTMCell::LSTMEvalFloat16( in LSTMEvalFloat16()
722 bool LSTMCell::LSTMStep( in LSTMStep()
952 bool LSTMCell::Eval() { in Eval()
|
| D | UnidirectionalSequenceLSTM.cpp | 379 LSTMCell::LSTMEvalFloat32( in execute()
420 LSTMCell::LSTMEvalFloat16( in execute()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1118 RunTimeOperandInfo& scratch = mOperands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1119 RunTimeOperandInfo& outputStateOut = mOperands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1120 RunTimeOperandInfo& cellStateOut = mOperands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1121 RunTimeOperandInfo& output = mOperands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1124 LSTMCell lstm_cell(operation, mOperands); in executeOperation()
|