| /packages/modules/NeuralNetworks/common/operations/ |
| 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
95 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in fullyConnectedFloat16()
100 weightsShape, biasDataFloat32.data(), biasShape, activation, in fullyConnectedFloat16()
109 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
122 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
140 weightsOffset, biasData, convertShapeToDims(biasShape), in fullyConnectedQuant8()
150 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
[all …]
|
| D | GroupedConv2D.cpp | 47 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32() argument
105 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8() argument
120 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in groupedConvQuant8()
183 const int32_t* biasData, const Shape& biasShape,
192 const int32_t* biasData, const Shape& biasShape,
203 const Shape& biasShape, int32_t padding_left, in groupedConvQuant8PerChannel() argument
220 Shape biasChannelShape = biasShape; in groupedConvQuant8PerChannel()
287 const _Float16* biasData, const Shape& biasShape, int32_t padding_left, in groupedConvFloat16() argument
295 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in groupedConvFloat16()
303 biasData_float32.data(), biasShape, padding_left, padding_right, padding_top, in groupedConvFloat16()
[all …]
|
| 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
243 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in convNhwc()
266 convertShapeToDims(biasShape), stride_width, stride_height, 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
316 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in convNhwc()
324 biasData_float32.data(), biasShape, padding_left, padding_right, padding_top, in convNhwc()
[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
179 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in depthwiseConvNhwc()
184 biasDataFloat32.data(), biasShape, paddingLeft, paddingRight, paddingTop, in depthwiseConvNhwc()
194 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
210 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in depthwiseConvNhwc()
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()
[all …]
|
| D | TransposeConv2D.cpp | 127 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument
184 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
210 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
279 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc() argument
285 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in transposeConvNhwc()
293 biasData_float32.data(), biasShape, param, outputData_float32.data(), 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()
320 const Shape& biasShape, const TransposeConv2dParam& param, in transposeConvQuant8PerChannelNhwc() argument
348 Shape biasChannelShape = biasShape; in transposeConvQuant8PerChannelNhwc()
[all …]
|
| D | QuantizedLSTM.cpp | 383 Shape biasShape; in eval() local
384 biasShape.dimensions = {getSizeOfDimension(weightsShape, 0)}; in eval()
385 std::vector<int32_t> bias(getNumberOfElements(biasShape)); in eval()
415 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
|
| /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,
76 const Shape& biasShape, int32_t paddingLeft,
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,
174 const Shape& biasShape, int32_t padding_left,
|
| D | OperationsUtils.h | 180 const Shape& biasShape, const Shape& outputShape,
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 257 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler() argument
261 const double bias_scale = biasShape.scale; in GetQuantizedConvolutionMultipler()
|