Home
last modified time | relevance | path

Searched refs:output_float32 (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTM.cpp631 std::vector<float> output_float32(maxTime * batchOutputSize); in LSTMEvalFloat16() local
632 convertFloat16ToFloat32(output_buffer, &output_float32); in LSTMEvalFloat16()
661 float* outputData = timeMajor ? output_float32.data() : transposedOutput.data(); in LSTMEvalFloat16()
711 output_float32.data()); in LSTMEvalFloat16()
716 convertFloat32ToFloat16(output_float32, output_buffer); in LSTMEvalFloat16()
DGenerateProposals.cpp113 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformFloat16() local
117 output_float32.data(), outputShape)); in bboxTransformFloat16()
118 convertFloat32ToFloat16(output_float32, outputData); in bboxTransformFloat16()
135 std::vector<float> output_float32(getNumberOfElements(outputShape)); in bboxTransformQuant() local
139 output_float32.data(), outputShape)); in bboxTransformQuant()
140 convertFloat32ToQuant(output_float32, outputShape.scale, outputShape.offset, outputData); in bboxTransformQuant()