Android NN API

// 2018-08-06 11:04
#include <NeuralNetworks.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <stdio.h>

int main(int argc, char *argv[]) {
    ANeuralNetworksModel* model = NULL;
    ANeuralNetworksModel_create(&model);

    ANeuralNetworksOperandType tensorW;
    tensorW.type = ANEURALNETWORKS_TENSOR_FLOAT32;
    tensorW.dimensionCount = 0;
    tensorW.dimensions = NULL;

    ANeuralNetworksOperandType tensorX;
    tensorX.type = ANEURALNETWORKS_TENSOR_FLOAT32;
    tensorX.dimensionCount = 0;
    tensorX.dimensions = NULL;

    ANeuralNetworksOperandType tensorOut;
    tensorOut.type = ANEURALNETWORKS_TENSOR_FLOAT32;
    tensorOut.dimensionCount = 0;
    tensorOut.dimensions = NULL;

    ANeuralNetworksOperandType activationType;
    activationType.type = ANEURALNETWORKS_INT32;
    activationType.scale = 0.f;
    activationType.zeroPoint = 0;
    activationType.dimensionCount = 0;
    activationType.dimensions = NULL;

    int ret = 0;

    ret += ANeuralNetworksModel_addOperand(model, &tensorW);
    ret += ANeuralNetworksModel_addOperand(model, &tensorX);
    ret += ANeuralNetworksModel_addOperand(model, &activationType);
    ret += ANeuralNetworksModel_addOperand(model, &tensorOut);

    // w=111.
    float w_value = 111.0;
    ret += ANeuralNetworksModel_setOperandValue(model, 0, &w_value, sizeof(w_value));

    int32_t noneValue = ANEURALNETWORKS_FUSED_NONE;
    ret += ANeuralNetworksModel_setOperandValue(model, 2, &noneValue, sizeof(noneValue));

    uint32_t addInputIndexes[3]= {0, 1, 2};
    uint32_t addOutputIndexes[1] = {3};
    ret += ANeuralNetworksModel_addOperation(model, ANEURALNETWORKS_ADD, 3, addInputIndexes, 1, addOutputIndexes);

    uint32_t modelInputIndexes[1] = {1};
    uint32_t modelOutputIndexes[1] = {3};
    ret += ANeuralNetworksModel_identifyInputsAndOutputs(model, 1, modelInputIndexes, 1, modelOutputIndexes);
    ret += ANeuralNetworksModel_finish(model);

    // compilation
    ANeuralNetworksCompilation* compilation;
    ret += ANeuralNetworksCompilation_create(model, &compilation);

    ANeuralNetworksCompilation_finish(compilation);

    // run
    ANeuralNetworksExecution* run = NULL;
    ret += ANeuralNetworksExecution_create(compilation, &run);

    float myInput = 10;
    ANeuralNetworksExecution_setInput(run, 0, NULL, &myInput, sizeof(myInput));

    float myOutput = 0;
    ret += ANeuralNetworksExecution_setOutput(run, 0, NULL, &myOutput, sizeof(myOutput));

    ANeuralNetworksEvent* run_end = NULL;
    ret += ANeuralNetworksExecution_startCompute(run, &run_end);

    ret += ANeuralNetworksEvent_wait(run_end);
    printf("%f\n", myOutput);

    ANeuralNetworksEvent_free(run_end);
    ANeuralNetworksExecution_free(run);
}

cc_binary {
    srcs: ["nn api_test.cpp"],
    cflags: ["-Wno-error-unused-parameter"],
    name: "nn api_test",
    include_dirs: ["frameworks/ml/nn/runtime/include"],
    shared_libs: ["libneuralnetworks"],
}