| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | FullyConnected.cpp | 54 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32() argument
68 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
75 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
84 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16() argument
92 convertFloat16ToFloat32(biasData, &biasDataFloat32); in fullyConnectedFloat16()
105 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
136 weightsOffset, biasData, convertShapeToDims(biasShape), in fullyConnectedQuant8()
146 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
175 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()
|
| D | GroupedConv2D.cpp | 51 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in groupedConvFloat32() argument
92 sum += biasData[g * outputGroupDepth + d]; in groupedConvFloat32()
109 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8() argument
167 sum += biasData[g * outputGroupDepth + d]; in groupedConvQuant8()
187 const int32_t* biasData, const Shape& biasShape,
196 const int32_t* biasData, const Shape& biasShape,
206 const float* filterScales, const int32_t* biasData, in groupedConvQuant8PerChannel() argument
272 sum += biasData[channelIndex]; in groupedConvQuant8PerChannel()
292 const _Float16* biasData, const Shape& biasShape, int32_t padding_left, in groupedConvFloat16() argument
305 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 | DepthwiseConv2D.cpp | 130 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
157 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc()
165 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvNhwc() argument
176 convertFloat16ToFloat32(biasData, &biasDataFloat32); in depthwiseConvNhwc()
190 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
233 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc()
241 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
261 filterShape, biasData, biasShape, paddingLeft, paddingRight, in depthwiseConvNhwc()
274 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc() argument
351 sum += biasData[oc]; in depthwiseConvQuant8PerChannelNhwc()
[all …]
|
| D | Conv2D.cpp | 192 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument
213 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
221 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument
262 convertShapeToDims(filterShape), filterOffset, biasData, in convNhwc()
275 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument
294 biasData, biasShape, padding_left, padding_right, padding_top, in convNhwc()
304 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc() argument
318 convertFloat16ToFloat32(biasData, &biasData_float32); in convNhwc()
331 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv() argument
341 biasData, biasShape, padding_left, padding_right, padding_top, in conv()
[all …]
|
| D | TransposeConv2D.cpp | 123 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in transposeConvNhwc() argument
170 *outPtr += biasData[d]; in transposeConvNhwc()
180 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
262 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvNhwc()
275 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc() argument
286 convertFloat16ToFloat32(biasData, &biasData_float32); in transposeConvNhwc()
298 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv() argument
306 filterShape, biasData, biasShape, param, output.getNhwcBuffer(), in transposeConv()
315 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument
403 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvQuant8PerChannelNhwc()
[all …]
|
| D | SVDF.cpp | 164 void SVDF::EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData, in EvalFloat32() argument
212 tflite::tensor_utils::VectorBatchVectorAdd(biasData, num_units, batch_size, outputData); in EvalFloat32()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 37 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft,
43 const Shape& filterShape, const float* biasData, const Shape& biasShape,
51 const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft,
58 const float* filterScales, const int32_t* biasData,
133 const _Float16* biasData, const Shape& biasShape, int32_t numGroups,
139 const Shape& filterShape, const float* biasData, const Shape& biasShape,
147 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
156 const float* filterScales, const int32_t* biasData,
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | RNN.h | 57 const T* biasData, const T* weightsData, const Shape& weightsShape,
64 const T* biasData, const T* weightsData, const Shape& weightsShape,
|
| D | SVDF.h | 59 void EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData,
|