Searched refs:outputData_float32 (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | Softmax.cpp | 106 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in softmaxFloat16() local
108 softmaxFloat32(inputData_float32.data(), inputShape, beta, axis, outputData_float32.data(), in softmaxFloat16()
110 convertFloat32ToFloat16(outputData_float32, outputData); in softmaxFloat16()
|
| D | ResizeImageOps.cpp | 144 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in resizeImageOpNhwc() local
146 halfPixelCenters, outputData_float32.data(), outputShape)); in resizeImageOpNhwc()
147 convertFloat32ToFloat16(outputData_float32, outputData); in resizeImageOpNhwc()
|
| D | GroupedConv2D.cpp | 296 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in groupedConvFloat16() local
305 outputData_float32.data(), outputShape); in groupedConvFloat16()
306 convertFloat32ToFloat16(outputData_float32, outputData); in groupedConvFloat16()
|
| D | Pooling.cpp | 245 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in maxPoolNhwc() local
248 maxPoolNhwc(inputData_float32.data(), inputShape, param, outputData_float32.data(), in maxPoolNhwc()
250 convertFloat32ToFloat16(outputData_float32, outputData); in maxPoolNhwc()
|
| D | TransposeConv2D.cpp | 286 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in transposeConvNhwc() local
293 biasData_float32.data(), biasShape, param, outputData_float32.data(), in transposeConvNhwc()
295 convertFloat32ToFloat16(outputData_float32, outputData); in transposeConvNhwc()
|
| D | Conv2D.cpp | 317 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in convNhwc() local
326 dilation_height_factor, activation, outputData_float32.data(), outputShape); in convNhwc()
327 convertFloat32ToFloat16(outputData_float32, outputData); in convNhwc()
|