Searched refs:QuantizedLSTMCell (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTMTest.cpp | 62 inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor].shape[1]; in QuantizedLSTMOpModel()
76 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kInputTensor], &input_); in QuantizedLSTMOpModel()
77 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kPrevOutputTensor], in QuantizedLSTMOpModel()
79 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor], in QuantizedLSTMOpModel()
96 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputTensor, input_), in invoke()
98 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToInputWeightsTensor, in invoke()
101 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToForgetWeightsTensor, in invoke()
104 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToCellWeightsTensor, in invoke()
107 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToOutputWeightsTensor, in invoke()
110 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
[all …]
|
| D | QuantizedLSTM.h | 29 class QuantizedLSTMCell {
31 QuantizedLSTMCell(const hal::Operation& operation, RunTimeOperandInfo* operands);
|
| D | QuantizedLSTM.cpp | 223 QuantizedLSTMCell::QuantizedLSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in QuantizedLSTMCell() function in android::nn::QuantizedLSTMCell
248 bool QuantizedLSTMCell::prepare(const Operation& operation, RunTimeOperandInfo* operands, in prepare()
352 void QuantizedLSTMCell::concatenateWeights(const std::vector<uint32_t>& weightsDims, in concatenateWeights()
368 void QuantizedLSTMCell::concatenateBiases(uint32_t outputSize, int32_t* bias) { in concatenateBiases()
377 bool QuantizedLSTMCell::eval() { in eval()
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1674 operands[outs[QuantizedLSTMCell::kCellStateOutTensor]]; in executeOperation()
1675 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation()
1678 QuantizedLSTMCell quantizedLSTMCell(operation, operands); in executeOperation()
1680 success = QuantizedLSTMCell::prepare(operation, operands, &cellStateOutShape, in executeOperation()
|