Searched refs:output_float32 (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| 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()
|
| D | GenerateProposals.cpp | 120 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformFloat16() local
124 output_float32.data(), outputShape)); in bboxTransformFloat16()
125 convertFloat32ToFloat16(output_float32, outputData); in bboxTransformFloat16()
142 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformQuant() local
146 output_float32.data(), outputShape)); in bboxTransformQuant()
147 convertFloat32ToQuant(output_float32, outputShape.scale, outputShape.offset, outputData); in bboxTransformQuant()
164 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformQuant() local
168 output_float32.data(), outputShape)); in bboxTransformQuant()
169 convertFloat32ToQuant(output_float32, outputShape.scale, outputShape.offset, outputData); in bboxTransformQuant()
|