Home
last modified time | relevance | path

Searched refs:recurrentToCellWeights (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/runtime/test/generated/models/
Dquantized_lstm.model.cpp18 auto recurrentToCellWeights = model->addOperand(&type2); in CreateModel() local
29 …utputWeights, recurrentToInputWeights, recurrentToForgetWeights, recurrentToCellWeights, recurrent… in CreateModel()
32 …utputWeights, recurrentToInputWeights, recurrentToForgetWeights, recurrentToCellWeights, recurrent… in CreateModel()
57 auto recurrentToCellWeights = model->addOperand(&type2); in CreateModel_relaxed() local
68 …utputWeights, recurrentToInputWeights, recurrentToForgetWeights, recurrentToCellWeights, recurrent… in CreateModel_relaxed()
71 …utputWeights, recurrentToInputWeights, recurrentToForgetWeights, recurrentToCellWeights, recurrent… in CreateModel_relaxed()
100 auto recurrentToCellWeights = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
111 …utputWeights, recurrentToInputWeights, recurrentToForgetWeights, recurrentToCellWeights, recurrent… in CreateModel_dynamic_output_shape()
114 …utputWeights, recurrentToInputWeights, recurrentToForgetWeights, recurrentToCellWeights, recurrent… in CreateModel_dynamic_output_shape()
141 auto recurrentToCellWeights = model->addOperand(&type2); in CreateModel_dynamic_output_shape_relaxed() local
[all …]
/frameworks/ml/nn/common/operations/
DQuantizedLSTMTest.cpp164 std::vector<uint8_t> recurrentToCellWeights, in setWeightsAndBiases() argument
176 recurrentToCellWeights_ = recurrentToCellWeights; in setWeightsAndBiases()
DQuantizedLSTM.cpp283 auto recurrentToCellWeights = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in prepare() local
287 NN_RET_CHECK(checkWeightsShape(recurrentToCellWeights, outputSize)); in prepare()
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp3104 ANeuralNetworksOperandType recurrentToCellWeights = recurrentToInputWeights; in unidirectionalSequenceLSTMTest() local
3190 recurrentToCellWeights, in unidirectionalSequenceLSTMTest()