interface.cpp

#if defined(_MSC_VER) && !defined(COMPILER_MSVC)
#  define COMPILER_MSVC // Set MSVC visibility of exported symbols in the shared library.
#endif
#include <c_api.h> // TensorFlow C API header
#include "tf_utils.hpp"
#include <array>
#include <iostream>
#include <algorithm>
#include <vector>

static void DeallocateTensor(void* data, size_t len, void* arg) {
  free(data);
  std::cout << "Deallocate tensor" << std::endl;
}

int main() {
  TF_Graph* graph = LoadGraphDef("graph.pb");
  if (graph == nullptr) {
    std::cout << "Can't load graph" << std::endl;
    return 1;
  }

  TF_Output input_op = {TF_GraphOperationByName(graph, "input_4"), 0};
  if (input_op.oper == nullptr) {
    std::cout << "Can't init input_op" << std::endl;
    return 2;
  }

  const std::array<int64_t, 3> input_dims = {1, 5, 12};
  std::size_t input_vals_size = sizeof(float);
  for (auto i : input_dims) {
    input_vals_size *= i;
  }
  auto input_vals = static_cast<float*>(malloc(input_vals_size));
  std::vector<float> m_input_vals = {
    -0.4809832f, -0.3770838f, 0.1743573f, 0.7720509f, -0.4064746f, 0.0116595f, 0.0051413f, 0.9135732f, 0.7197526f, -0.0400658f, 0.1180671f, -0.6829428f,
    -0.4810135f, -0.3772099f, 0.1745346f, 0.7719303f, -0.4066443f, 0.0114614f, 0.0051195f, 0.9135003f, 0.7196983f, -0.0400035f, 0.1178188f, -0.6830465f,
    -0.4809143f, -0.3773398f, 0.1746384f, 0.7719052f, -0.4067171f, 0.0111654f, 0.0054433f, 0.9134697f, 0.7192584f, -0.0399981f, 0.1177435f, -0.6835230f,
    -0.4808300f, -0.3774327f, 0.1748246f, 0.7718700f, -0.4070232f, 0.0109549f, 0.0059128f, 0.9133330f, 0.7188759f, -0.0398740f, 0.1181437f, -0.6838635f,
    -0.4807833f, -0.3775733f, 0.1748378f, 0.7718275f, -0.4073670f, 0.0107582f, 0.0062978f, 0.9131795f, 0.7187147f, -0.0394935f, 0.1184392f, -0.6840039f,
  };

  std::copy(m_input_vals.begin(), m_input_vals.end(), input_vals); // init input_vals.

  TF_Tensor* input_tensor = TF_NewTensor(TF_FLOAT,
                                         input_dims.data(), static_cast<int>(input_dims.size()),
                                         input_vals, input_vals_size,
                                         DeallocateTensor, nullptr);

  TF_Output out_op = {TF_GraphOperationByName(graph, "output_node0"), 0};
  if (input_op.oper == nullptr) {
    std::cout << "Can't init out_op" << std::endl;
    return 3;
  }

  TF_Tensor* output_tensor = nullptr;

  bool success = RunSession(graph,
                            &input_op, &input_tensor, 1,
                            &out_op, &output_tensor, 1);

  if (success) {
    auto data = static_cast<float*>(TF_TensorData(output_tensor));
    std::cout << "Output vals: " << data[0] << "," << data[1] << "," << data[2] << "," << data[3] << std::endl;
  } else {
    std::cout << "Error run session";
    return 4;
  }

  TF_DeleteTensor(input_tensor);
  TF_DeleteTensor(output_tensor);
  TF_DeleteGraph(graph);

  return 0;
}