Home
last modified time | relevance | path

Searched refs:biasShape (Results 1 – 9 of 9) sorted by relevance

/frameworks/ml/nn/common/operations/
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
94 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in fullyConnectedFloat16()
99 weightsShape, biasDataFloat32.data(), biasShape, activation, in fullyConnectedFloat16()
108 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
121 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, weightsShape, biasShape, outputShape, in fullyConnectedQuant8()
139 weightsOffset, biasData, convertShapeToDims(biasShape), in fullyConnectedQuant8()
149 const int32_t* biasData, const Shape& biasShape, int32_t activation, in fullyConnectedQuant8() argument
[all …]
DGroupedConv2D.cpp47 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 …]
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
242 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in convNhwc()
265 convertShapeToDims(biasShape), stride_width, stride_height, 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
315 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in convNhwc()
323 biasData_float32.data(), biasShape, padding_left, padding_right, padding_top, in convNhwc()
[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
177 std::vector<float> biasDataFloat32(getNumberOfElements(biasShape)); in depthwiseConvNhwc()
182 biasDataFloat32.data(), biasShape, paddingLeft, paddingRight, paddingTop, in depthwiseConvNhwc()
192 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc() argument
208 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in depthwiseConvNhwc()
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()
[all …]
DTransposeConv2D.cpp126 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc() argument
183 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc() argument
209 NN_RET_CHECK(GetQuantizedConvolutionMultipler(inputShape, filterShape, biasShape, outputShape, in transposeConvNhwc()
278 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc() argument
284 std::vector<float> biasData_float32(getNumberOfElements(biasShape)); in transposeConvNhwc()
292 biasData_float32.data(), biasShape, param, outputData_float32.data(), 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()
319 const Shape& biasShape, const TransposeConv2dParam& param, in transposeConvQuant8PerChannelNhwc() argument
347 Shape biasChannelShape = biasShape; in transposeConvQuant8PerChannelNhwc()
[all …]
DQuantizedLSTM.cpp386 Shape biasShape; in eval() local
387 biasShape.dimensions = {getSizeOfDimension(weightsShape, 0)}; in eval()
388 std::vector<int32_t> bias(getNumberOfElements(biasShape)); in eval()
418 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
/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,
72 const Shape& biasShape, int32_t paddingLeft,
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,
170 const Shape& biasShape, int32_t padding_left,
DOperationsUtils.h196 const Shape& biasShape, const Shape& outputShape,
/frameworks/ml/nn/common/
DOperationsUtils.cpp259 const Shape& biasShape, const Shape& outputShape, in GetQuantizedConvolutionMultipler() argument
263 const double bias_scale = biasShape.scale; in GetQuantizedConvolutionMultipler()