Lines Matching refs:weightsShape
56 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()
148 const int8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8() argument
159 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
167 params.weights_offset = -weightsShape.offset; in fullyConnectedQuant8()
177 convertShapeToTflshape(weightsShape), weightsData, convertShapeToTflshape(biasShape), in fullyConnectedQuant8()