Searched refs:output_float32 (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | GenerateProposals.cpp | 124 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformFloat16() local
128 output_float32.data(), outputShape)); in bboxTransformFloat16()
129 convertFloat32ToFloat16(output_float32, outputData); in bboxTransformFloat16()
146 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformQuant() local
150 output_float32.data(), outputShape)); in bboxTransformQuant()
151 convertFloat32ToQuant(output_float32, outputShape.scale, outputShape.offset, outputData); in bboxTransformQuant()
168 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformQuant() local
172 output_float32.data(), outputShape)); in bboxTransformQuant()
173 convertFloat32ToQuant(output_float32, outputShape.scale, outputShape.offset, outputData); in bboxTransformQuant()
|
| D | LSTM.cpp | 674 std::vector<float> output_float32(maxTime * batchOutputSize); in LSTMEvalFloat16() local
675 convertFloat16ToFloat32(output_buffer, &output_float32); in LSTMEvalFloat16()
704 float* outputData = timeMajor ? output_float32.data() : transposedOutput.data(); in LSTMEvalFloat16()
754 output_float32.data()); in LSTMEvalFloat16()
759 convertFloat32ToFloat16(output_float32, output_buffer); in LSTMEvalFloat16()
|