/packages/modules/NeuralNetworks/common/operations/ |
D | GroupedConv2D.cpp | 47 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32() argument 88 sum += biasData[g * outputGroupDepth + d]; in groupedConvFloat32() 105 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8() argument 163 sum += biasData[g * outputGroupDepth + d]; in groupedConvQuant8() 183 const int32_t* biasData, const Shape& biasShape, 192 const int32_t* biasData, const Shape& biasShape, 202 const float* filterScales, const int32_t* biasData, in groupedConvQuant8PerChannel() argument 267 sum += biasData[channelIndex]; in groupedConvQuant8PerChannel() 287 const _Float16* biasData, const Shape& biasShape, int32_t padding_left, in groupedConvFloat16() argument 300 convertFloat16ToFloat32(biasData, &biasData_float32); in groupedConvFloat16() [all …]
|
D | RNN.cpp | 117 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep() argument 125 hiddenStateInputData, biasData, weightsData, weightsShape, in RNNStep() 137 const Shape& auxInputShape, const T* hiddenStateInputData, const T* biasData, in RNNStep() argument 180 output_ptr_batch[o] = biasData[o]; in RNNStep() 224 const _Float16* hiddenStateInputData, const _Float16* biasData, 231 const _Float16* hiddenStateInputData, const _Float16* biasData, 240 const float* hiddenStateInputData, const float* biasData, 247 const float* hiddenStateInputData, const float* biasData,
|
D | FullyConnected.cpp | 58 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32() argument 72 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32() 79 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32() 88 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16() argument 96 convertFloat16ToFloat32(biasData, &biasDataFloat32); in fullyConnectedFloat16() 109 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument 140 weightsOffset, biasData, convertShapeToDims(biasShape), in fullyConnectedQuant8() 150 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument 179 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()
|
D | Conv2D.cpp | 195 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument 216 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc() 224 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument 265 convertShapeToDims(filterShape), filterOffset, biasData, in convNhwc() 278 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument 297 biasData, biasShape, padding_left, padding_right, padding_top, in convNhwc() 307 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc() argument 321 convertFloat16ToFloat32(biasData, &biasData_float32); in convNhwc() 334 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv() argument 344 biasData, biasShape, padding_left, padding_right, padding_top, in conv() [all …]
|
D | DepthwiseConv2D.cpp | 134 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument 161 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc() 169 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvNhwc() argument 180 convertFloat16ToFloat32(biasData, &biasDataFloat32); in depthwiseConvNhwc() 194 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument 237 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc() 245 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument 265 filterShape, biasData, biasShape, paddingLeft, paddingRight, in depthwiseConvNhwc() 278 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc() argument 355 sum += biasData[oc]; in depthwiseConvQuant8PerChannelNhwc() [all …]
|
D | TransposeConv2D.cpp | 127 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument 174 *outPtr += biasData[d]; in transposeConvNhwc() 184 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument 266 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvNhwc() 279 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc() argument 290 convertFloat16ToFloat32(biasData, &biasData_float32); in transposeConvNhwc() 302 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv() argument 310 filterShape, biasData, biasShape, param, output.getNhwcBuffer(), in transposeConv() 319 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument 407 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvQuant8PerChannelNhwc() [all …]
|
D | RNN.h | 51 const T* biasData, const T* weightsData, const Shape& weightsShape, 58 const T* biasData, const T* weightsData, const Shape& weightsShape,
|
D | SVDF.cpp | 162 void SVDF::EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData, in EvalFloat32() argument 210 tflite::tensor_utils::VectorBatchVectorAdd(biasData, num_units, batch_size, outputData); in EvalFloat32()
|
D | SVDF.h | 59 void EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData,
|
/packages/modules/NeuralNetworks/common/include/ |
D | Operations.h | 54 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, 60 const Shape& filterShape, const float* biasData, const Shape& biasShape, 68 const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft, 75 const float* filterScales, const int32_t* biasData, 150 const _Float16* biasData, const Shape& biasShape, int32_t numGroups, 156 const Shape& filterShape, const float* biasData, const Shape& biasShape, 164 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, 173 const float* filterScales, const int32_t* biasData,
|