Searched refs:outputData_float32 (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Softmax.cpp | 104 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in softmaxFloat16() local
106 softmaxFloat32(inputData_float32.data(), inputShape, beta, axis, outputData_float32.data(), in softmaxFloat16()
108 convertFloat32ToFloat16(outputData_float32, outputData); in softmaxFloat16()
|
| D | ResizeImageOps.cpp | 142 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in resizeImageOpNhwc() local
144 halfPixelCenters, outputData_float32.data(), outputShape)); in resizeImageOpNhwc()
145 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 | 243 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in maxPoolNhwc() local
246 maxPoolNhwc(inputData_float32.data(), inputShape, param, outputData_float32.data(), in maxPoolNhwc()
248 convertFloat32ToFloat16(outputData_float32, outputData); in maxPoolNhwc()
|
| D | TransposeConv2D.cpp | 285 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in transposeConvNhwc() local
292 biasData_float32.data(), biasShape, param, outputData_float32.data(), in transposeConvNhwc()
294 convertFloat32ToFloat16(outputData_float32, outputData); in transposeConvNhwc()
|
| D | Conv2D.cpp | 316 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in convNhwc() local
325 dilation_height_factor, activation, outputData_float32.data(), outputShape); in convNhwc()
326 convertFloat32ToFloat16(outputData_float32, outputData); in convNhwc()
|