Searched refs:cellGateBias (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | quantized_lstm.model.cpp | 22 auto cellGateBias = model->addOperand(&type3); in CreateModel() local
29 …CellWeights, recurrentToOutputWeights, inputGateBias, forgetGateBias, cellGateBias, outputGateBias… in CreateModel()
32 …CellWeights, recurrentToOutputWeights, inputGateBias, forgetGateBias, cellGateBias, outputGateBias… in CreateModel()
61 auto cellGateBias = model->addOperand(&type3); in CreateModel_relaxed() local
68 …CellWeights, recurrentToOutputWeights, inputGateBias, forgetGateBias, cellGateBias, outputGateBias… in CreateModel_relaxed()
71 …CellWeights, recurrentToOutputWeights, inputGateBias, forgetGateBias, cellGateBias, outputGateBias… in CreateModel_relaxed()
104 auto cellGateBias = model->addOperand(&type3); in CreateModel_dynamic_output_shape() local
111 …CellWeights, recurrentToOutputWeights, inputGateBias, forgetGateBias, cellGateBias, outputGateBias… in CreateModel_dynamic_output_shape()
114 …CellWeights, recurrentToOutputWeights, inputGateBias, forgetGateBias, cellGateBias, outputGateBias… in CreateModel_dynamic_output_shape()
145 auto cellGateBias = model->addOperand(&type3); in CreateModel_dynamic_output_shape_relaxed() local
[all …]
|
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTMTest.cpp | 168 std::vector<int32_t> cellGateBias, // in setWeightsAndBiases() argument
180 cellGateBias_ = cellGateBias; in setWeightsAndBiases()
|
| D | QuantizedLSTM.cpp | 305 auto cellGateBias = GetInput(operation, operands, kCellGateBiasTensor); in prepare() local
309 NN_RET_CHECK(checkBiasShape(cellGateBias)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3119 ANeuralNetworksOperandType cellGateBias = inputGateBias; in unidirectionalSequenceLSTMTest() local
3197 cellGateBias, in unidirectionalSequenceLSTMTest()
|