Searched refs:weightsData (Results 1 – 4 of 4) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | RNN.cpp | 117 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep() argument
125 hiddenStateInputData, biasData, weightsData, weightsShape, in RNNStep()
138 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep() argument
171 const T* input_weights_ptr = weightsData; in RNNStep()
225 const _Float16* weightsData, const Shape& weightsShape,
232 const _Float16* weightsData, const Shape& weightsShape,
241 const float* weightsData, const Shape& weightsShape,
248 const float* weightsData, const Shape& weightsShape,
|
| H A D | FullyConnected.cpp | 57 const float* weightsData, const Shape& weightsShape, in fullyConnectedFloat32() argument
71 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
78 weightsData, convertShapeToDims(weightsShape), in fullyConnectedFloat32()
87 const _Float16* weightsData, const Shape& weightsShape, in fullyConnectedFloat16() argument
94 convertFloat16ToFloat32(weightsData, &weightsDataFloat32); in fullyConnectedFloat16()
108 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8() argument
139 weightsData, convertShapeToDims(weightsShape), in fullyConnectedQuant8()
149 const int8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8() argument
178 convertShapeToTflshape(weightsShape), weightsData, convertShapeToTflshape(biasShape), in fullyConnectedQuant8()
|
| H A D | RNN.h | 51 const T* biasData, const T* weightsData, const Shape& weightsShape,
58 const T* biasData, const T* weightsData, const Shape& weightsShape,
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestMemoryInternal.cpp | 108 uint8_t* weightsData = in TEST_F() local
110 ASSERT_NE(weightsData, nullptr); in TEST_F()
111 memcpy(weightsData + offsetForMatrix2, matrix2, sizeof(matrix2)); in TEST_F()
112 memcpy(weightsData + offsetForMatrix3, matrix3, sizeof(matrix3)); in TEST_F()
172 munmap(weightsData, weightsSize); in TEST_F()
|