Searched refs:LSTMCell (Results 1 – 7 of 7) sorted by relevance
/packages/modules/NeuralNetworks/common/cpu_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 | 199 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 …]
|
D | BidirectionalSequenceLSTM.cpp | 287 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()
|
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() 992 bool LSTMCell::Eval() { in Eval()
|
D | UnidirectionalSequenceLSTM.cpp | 316 LSTMCell::LSTMEvalFloat32( in execute() 368 LSTMCell::LSTMEvalFloat16( in execute()
|
/packages/modules/NeuralNetworks/common/types/operations/include/ |
D | LSTM.h | 48 class LSTMCell { 50 LSTMCell(const Operation& operation, RunTimeOperandInfo* operands);
|
/packages/modules/NeuralNetworks/common/ |
D | CpuExecutor.cpp | 1042 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()
|