Home
last modified time | relevance | path

Searched refs:biasData (Results 1 – 10 of 10) sorted by relevance

/frameworks/ml/nn/common/operations/
DRNN.cpp121 const T* biasData, const T* weightsData, const Shape& weightsShape, in RNNStep() argument
129 hiddenStateInputData, biasData, weightsData, weightsShape, in RNNStep()
141 const Shape& auxInputShape, const T* hiddenStateInputData, const T* biasData, in RNNStep() argument
184 output_ptr_batch[o] = biasData[o]; in RNNStep()
228 const _Float16* hiddenStateInputData, const _Float16* biasData,
235 const _Float16* hiddenStateInputData, const _Float16* biasData,
244 const float* hiddenStateInputData, const float* biasData,
251 const float* hiddenStateInputData, const float* biasData,
DGroupedConv2D.cpp47 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 …]
DFullyConnected.cpp57 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32() argument
71 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
78 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
87 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16() argument
95 convertFloat16ToFloat32(biasData, &biasDataFloat32); in fullyConnectedFloat16()
108 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
139 weightsOffset, biasData, convertShapeToDims(biasShape), in fullyConnectedQuant8()
149 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
178 biasData, convertShapeToTflshape(outputShape), outputData); in fullyConnectedQuant8()
DConv2D.cpp194 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument
215 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
223 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument
264 convertShapeToDims(filterShape), filterOffset, biasData, in convNhwc()
277 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument
296 biasData, biasShape, padding_left, padding_right, padding_top, in convNhwc()
306 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc() argument
320 convertFloat16ToFloat32(biasData, &biasData_float32); in convNhwc()
333 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv() argument
343 biasData, biasShape, padding_left, padding_right, padding_top, in conv()
[all …]
DDepthwiseConv2D.cpp132 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
159 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc()
167 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvNhwc() argument
178 convertFloat16ToFloat32(biasData, &biasDataFloat32); in depthwiseConvNhwc()
192 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
235 convertShapeToTflshape(biasShape), biasData, in depthwiseConvNhwc()
243 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
263 filterShape, biasData, biasShape, paddingLeft, paddingRight, in depthwiseConvNhwc()
276 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc() argument
353 sum += biasData[oc]; in depthwiseConvQuant8PerChannelNhwc()
[all …]
DTransposeConv2D.cpp126 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument
173 *outPtr += biasData[d]; in transposeConvNhwc()
183 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
265 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvNhwc()
278 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc() argument
289 convertFloat16ToFloat32(biasData, &biasData_float32); in transposeConvNhwc()
301 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv() argument
309 filterShape, biasData, biasShape, param, output.getNhwcBuffer(), in transposeConv()
318 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument
406 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvQuant8PerChannelNhwc()
[all …]
DRNN.h51 const T* biasData, const T* weightsData, const Shape& weightsShape,
58 const T* biasData, const T* weightsData, const Shape& weightsShape,
DSVDF.h59 void EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData,
DSVDF.cpp164 void SVDF::EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData, in EvalFloat32() argument
207 tflite::tensor_utils::VectorBatchVectorAssign(biasData, num_units, batch_size, outputData); in EvalFloat32()
/frameworks/ml/nn/common/include/
DOperations.h50 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft,
56 const Shape& filterShape, const float* biasData, const Shape& biasShape,
64 const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft,
71 const float* filterScales, const int32_t* biasData,
146 const _Float16* biasData, const Shape& biasShape, int32_t numGroups,
152 const Shape& filterShape, const float* biasData, const Shape& biasShape,
160 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
169 const float* filterScales, const int32_t* biasData,