| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 30 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvFloat16() argument
40 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in depthwiseConvFloat16()
45 biasDataFloat32.data(), biasShape, paddingLeft, paddingRight, paddingTop, 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
115 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in depthwiseConvQuant8()
141 convertShapeToTflshape(biasShape), biasData, in depthwiseConvQuant8()
149 const Shape& biasShape, int32_t paddingLeft, in depthwiseConvQuant8PerChannel() argument
182 Shape biasChannelShape = biasShape; 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
86 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in fullyConnectedFloat16()
91 weightsShape, biasDataFloat32.data(), biasShape, activation, in fullyConnectedFloat16()
100 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
113 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
131 weightsOffset, biasData, convertShapeToDims(biasShape), in fullyConnectedQuant8()
|
| D | GroupedConv2D.cpp | 43 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32() argument
100 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8() argument
115 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in groupedConvQuant8()
179 const Shape& biasShape, int32_t padding_left, in groupedConvQuant8PerChannel() argument
197 Shape biasChannelShape = biasShape; in groupedConvQuant8PerChannel()
264 const _Float16* biasData, const Shape& biasShape, int32_t padding_left, in groupedConvFloat16() argument
272 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in groupedConvFloat16()
280 biasData_float32.data(), biasShape, padding_left, padding_right, padding_top, in groupedConvFloat16()
|
| D | Conv2D.cpp | 165 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() argument
182 convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
190 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() argument
209 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in convNhwc()
228 convertShapeToDims(filterShape), filterOffset, biasData, convertShapeToDims(biasShape), in convNhwc()
237 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc() argument
246 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in convNhwc()
254 biasData_float32.data(), biasShape, padding_left, padding_right, padding_top, 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()
[all …]
|
| D | TransposeConv2D.cpp | 116 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument
172 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
199 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
268 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc() argument
274 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in transposeConvNhwc()
282 biasData_float32.data(), biasShape, param, outputData_float32.data(), 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()
308 const Shape& biasShape, const TransposeConv2dParam& param, in transposeConvQuant8PerChannelNhwc() argument
336 Shape biasChannelShape = biasShape; in transposeConvQuant8PerChannelNhwc()
[all …]
|
| D | QuantizedLSTM.cpp | 381 Shape biasShape; in eval() local
382 biasShape.dimensions = {getSizeOfDimension(weightsShape, 0)}; in eval()
383 std::vector<int32_t> bias(getNumberOfElements(biasShape)); in eval()
413 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
|
| /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,
73 const Shape& biasShape, int32_t paddingLeft,
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,
165 const Shape& biasShape, int32_t padding_left,
|
| D | OperationsUtils.h | 192 const Shape& biasShape, const Shape& outputShape,
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 214 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler() argument
218 const double bias_scale = biasShape.scale; in GetQuantizedConvolutionMultipler()
|