freedomtan / gldelegatebench Goto Github PK

private Object[] allocateInputBuffers(int[] shapes){
            int i_size = shapes.length;
            Object inputs[] = new Object[i_size];
            for (int i=0; i < i_size; i++) {
                ByteBuffer i_bytes = ByteBuffer.allocate(shapes[i]);
                float[] floats = new float[shapes[i] / 4];
                for(int j=0; j<floats.length; ++j) {
                    // Filling uniformly -5. ... 5.
                    floats[j] = ((float)j * 10) / floats.length - 5;
                }
                FloatBuffer fb = i_bytes.asFloatBuffer();
                fb.put(floats);
                i_bytes.rewind();
                Log.i("here: ", Float.toString(i_bytes.getFloat()) + " " + Float.toString(i_bytes.getFloat(4)));
                i_bytes.rewind();
                inputs[i] = i_bytes;
            }
            return inputs;
        }

String debugMessage = "";

            for (int i=0; i < loops; i++) {
                Object inputs[] = allocateInputBuffers(mModel.getInputShapes());
                Map<Integer, Object> outputs = allocateOutputBuffers(mModel.getOutputShapes());

                startTime = System.currentTimeMillis();
                interpreter.runForMultipleInputsOutputs(inputs, outputs);
                stopTime = System.currentTimeMillis();
                accTime += (stopTime - startTime);

                if(i == loops - 1) {
                    // on last step adding first element of output buffer to debugMessage
                    for (Map.Entry<Integer, Object> entry : outputs.entrySet()) {
                        ByteBuffer bb = (ByteBuffer) entry.getValue();
                        Log.i("here: ", "byteBufferLimit is : " + ((ByteBuffer) entry.getValue()).limit());
                        debugMessage += "    " + Float.toString(bb.getFloat(0)) + "\n";
                    }
                }
            }
            Log.i("here: ", "time: " + accTime/loops);
            resultMessage.setText("avg time: " + accTime/loops + " ms\n"
                + debugMessage);