Home
last modified time | relevance | path

Searched refs:inputToInputWeights (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.cpp264 auto inputToInputWeights = GetInput(operation, operands, kInputToInputWeightsTensor); in prepare() local
265 const float weightsScale = inputToInputWeights->scale; in prepare()
267 const float weightsZeroPoint = inputToInputWeights->zeroPoint; in prepare()
281 NN_RET_CHECK(checkWeightsShape(inputToInputWeights, inputSize)); in prepare()
DQuantizedLSTMTest.cpp158 void setWeightsAndBiases(std::vector<uint8_t> inputToInputWeights, in setWeightsAndBiases() argument
170 inputToInputWeights_ = inputToInputWeights; in setWeightsAndBiases()
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp4089 ANeuralNetworksOperandType inputToInputWeights = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest() local
4094 ANeuralNetworksOperandType inputToForgetWeights = inputToInputWeights; in unidirectionalSequenceLSTMTest()
4095 ANeuralNetworksOperandType inputToCellWeights = inputToInputWeights; in unidirectionalSequenceLSTMTest()
4096 ANeuralNetworksOperandType inputToOutputWeights = inputToInputWeights; in unidirectionalSequenceLSTMTest()
4183 inputToInputWeights, in unidirectionalSequenceLSTMTest()