Searched refs:weightsData (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNN.cpp | 121 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep() argument
129 hiddenStateInputData, biasData, weightsData, weightsShape, in RNNStep()
142 const T* weightsData, const Shape& weightsShape, const T* auxWeightsData, in RNNStep() argument
175 const T* input_weights_ptr = weightsData; 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,
|
| D | FullyConnected.cpp | 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
93 convertFloat16ToFloat32(weightsData, &weightsDataFloat32); in fullyConnectedFloat16()
107 const uint8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8() argument
138 weightsData, convertShapeToDims(weightsShape), in fullyConnectedQuant8()
148 const int8_t* weightsData, const Shape& weightsShape, in fullyConnectedQuant8() argument
177 convertShapeToTflshape(weightsShape), weightsData, convertShapeToTflshape(biasShape), in fullyConnectedQuant8()
|
| D | RNN.h | 51 const T* biasData, const T* weightsData, const Shape& weightsShape,
58 const T* biasData, const T* weightsData, const Shape& weightsShape,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestMemoryInternal.cpp | 109 uint8_t* weightsData = in TEST_F() local
111 ASSERT_NE(weightsData, nullptr); in TEST_F()
112 memcpy(weightsData + offsetForMatrix2, matrix2, sizeof(matrix2)); in TEST_F()
113 memcpy(weightsData + offsetForMatrix3, matrix3, sizeof(matrix3)); in TEST_F()
173 munmap(weightsData, weightsSize); in TEST_F()
|