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()
|