Searched defs:weightsShape (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | FullyConnected.cpp | 48 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
78 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
99 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
|
| D | RNN.cpp | 121 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep()
142 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep()
|
| D | UnidirectionalSequenceRNN.cpp | 61 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped() local
|
| D | QuantizedLSTM.cpp | 375 Shape weightsShape; in eval() local
|