Home
last modified time | relevance | path

Searched refs:weightsShape (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/operations/
DFullyConnected.cpp56 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32() argument
70 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
77 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
86 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16() argument
92 std::vector<float> weightsDataFloat32(getNumberOfElements(weightsShape)); in fullyConnectedFloat16()
99 weightsShape, biasDataFloat32.data(), biasShape, activation, in fullyConnectedFloat16()
107 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8() argument
112 int32_t weightsOffset = -weightsShape.offset; in fullyConnectedQuant8()
121 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
138 weightsData, convertShapeToDims(weightsShape), in fullyConnectedQuant8()
[all …]
DRNN.cpp121 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep() argument
127 uint32_t numUnits = weightsShape.dimensions[0]; in RNNStep()
129 hiddenStateInputData, biasData, weightsData, weightsShape, in RNNStep()
142 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep() argument
150 const uint32_t num_units = weightsShape.dimensions[0]; in RNNStep()
152 const uint32_t input_weights_stride = weightsShape.dimensions[1]; in RNNStep()
229 const _Float16* weightsData, const Shape& weightsShape,
236 const _Float16* weightsData, const Shape& weightsShape,
245 const float* weightsData, const Shape& weightsShape,
252 const float* weightsData, const Shape& weightsShape,
DQuantizedLSTM.cpp380 Shape weightsShape; in eval() local
381 weightsShape.dimensions = {4 * SizeOfDimension(prevOutput_, 1), in eval()
383 std::vector<uint8_t> weights(getNumberOfElements(weightsShape)); in eval()
384 concatenateWeights(weightsShape.dimensions, weights.data()); in eval()
387 biasShape.dimensions = {getSizeOfDimension(weightsShape, 0)}; in eval()
392 concatTempShape.dimensions = {SizeOfDimension(input_, 0), getSizeOfDimension(weightsShape, 1)}; in eval()
396 getSizeOfDimension(weightsShape, 0)}; in eval()
417 weights.data(), convertShapeToDims(weightsShape), bias.data(), in eval()
DRNN.h51 const T* biasData, const T* weightsData, const Shape& weightsShape,
58 const T* biasData, const T* weightsData, const Shape& weightsShape,
DUnidirectionalSequenceRNN.cpp70 Shape weightsShape = context->getInputShape(kWeightsTensor); in executeTyped() local
100 const uint32_t numUnits = getSizeOfDimension(weightsShape, 0); in executeTyped()
109 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()