| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 51 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft,
57 const Shape& filterShape, const float* biasData, const Shape& biasShape,
65 const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft,
72 const float* filterScales, const int32_t* biasData,
143 const _Float16* biasData, const Shape& biasShape, int32_t numGroups,
149 const Shape& filterShape, const float* biasData, const Shape& biasShape,
156 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
164 const float* filterScales, const int32_t* biasData,
|
| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 30 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvFloat16() argument
41 convertFloat16ToFloat32(biasData, &biasDataFloat32); in depthwiseConvFloat16()
66 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvFloat32() argument
92 convertShapeToTflshape(biasShape), biasData, in depthwiseConvFloat32()
100 const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvQuant8() argument
141 convertShapeToTflshape(biasShape), biasData, in depthwiseConvQuant8()
148 const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannel() argument
227 sum += biasData[oc]; in depthwiseConvQuant8PerChannel()
|
| D | FullyConnected.cpp | 49 const float* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat32() argument
63 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
70 biasData, convertShapeToDims(biasShape), in fullyConnectedFloat32()
79 const _Float16* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedFloat16() argument
87 convertFloat16ToFloat32(biasData, &biasDataFloat32); in fullyConnectedFloat16()
100 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
131 weightsOffset, biasData, convertShapeToDims(biasShape), in fullyConnectedQuant8()
|
| D | GroupedConv2D.cpp | 43 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32() argument
84 sum += biasData[g * outputGroupDepth + d]; in groupedConvFloat32()
100 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8() argument
158 sum += biasData[g * outputGroupDepth + d]; in groupedConvQuant8()
178 const float* filterScales, const int32_t* biasData, in groupedConvQuant8PerChannel() argument
244 sum += biasData[channelIndex]; in groupedConvQuant8PerChannel()
264 const _Float16* biasData, const Shape& biasShape, int32_t padding_left, in groupedConvFloat16() argument
277 convertFloat16ToFloat32(biasData, &biasData_float32); in groupedConvFloat16()
|
| D | TransposeConv2D.cpp | 116 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument
163 *outPtr += biasData[d]; in transposeConvNhwc()
172 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
255 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvNhwc()
268 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc() argument
279 convertFloat16ToFloat32(biasData, &biasData_float32); in transposeConvNhwc()
291 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv() argument
299 filterShape, biasData, biasShape, param, output.getNhwcBuffer(), in transposeConv()
307 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument
395 int32_t outVal = *bufferPtr + biasData[d]; in transposeConvQuant8PerChannelNhwc()
[all …]
|
| D | RNN.h | 49 const T* biasData, const T* weightsData, const Shape& weightsShape,
56 const T* biasData, const T* weightsData, const Shape& weightsShape,
|
| D | Conv2D.cpp | 165 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument
181 convertShapeToDims(filterShape), biasData, in convNhwc()
190 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument
228 convertShapeToDims(filterShape), filterOffset, biasData, convertShapeToDims(biasShape), in convNhwc()
237 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc() argument
251 convertFloat16ToFloat32(biasData, &biasData_float32); in convNhwc()
264 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv() argument
274 biasData, biasShape, padding_left, padding_right, padding_top, in conv()
284 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
358 sum += biasData[d]; in convQuant8PerChannelNhwc()
[all …]
|
| D | RNN.cpp | 121 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()
|
| D | SVDF.h | 57 void EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData,
|
| D | SVDF.cpp | 148 void SVDF::EvalFloat32(const float* inputData, const float* inputStateData, const float* biasData, in EvalFloat32() argument
191 tflite::tensor_utils::VectorBatchVectorAssign(biasData, num_units, batch_size, outputData); in EvalFloat32()
|