Home
last modified time | relevance | path

Searched defs:weightsShape (Results 1 – 4 of 4) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DFullyConnected.cpp57 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
87 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
108 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
149 const int8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
H A DRNN.cpp117 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep()
138 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep()
H A DUnidirectionalSequenceRNN.cpp69 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped() local
H A DQuantizedLSTM.cpp377 Shape weightsShape; in eval() local