Searched defs:weightsShape (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | FullyConnected.cpp | 53 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32()
83 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16()
104 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
145 const int8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8()
|
| D | RNN.cpp | 117 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep()
138 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep()
|
| D | UnidirectionalSequenceRNN.cpp | 57 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped() local
|
| D | QuantizedLSTM.cpp | 381 Shape weightsShape; in eval() local
|