Method body

plugins/wasmedge_tensorflow/tensorflow_func.cpp:73–115

} // namespace

Expect<uint32_t> CreateSession::body(const Runtime::CallingFrame &Frame,
                                     uint32_t ModBufPtr, uint32_t ModBufLen,
                                     uint32_t SessionIdPtr) {
  // Check memory instance from module.
  MEMINST_CHECK(MemInst, Frame, 0)

  // Check the input model buffer.
  MEM_SPAN_CHECK(ModBufSpan, MemInst, char, ModBufPtr, ModBufLen,
                 "Failed when accessing the input model buffer memory."sv)

  // Check the return value: SessionIdPtr should be valid.
  MEM_PTR_CHECK(SessionId, MemInst, uint32_t, SessionIdPtr,
                "Failed when accessing the return SessionID memory."sv)

  // Create context and import graph.
  uint32_t NewID = Env.newContext();
  SESSION_CHECK(Cxt, NewID, "Failed when allocating resources."sv,
                ErrNo::MissingMemory)

  Cxt->Graph = TF_NewGraph();
  Cxt->Buffer = TF_NewBufferFromString(ModBufSpan.data(), ModBufLen);
  Cxt->GraphOpts = TF_NewImportGraphDefOptions();
  TF_GraphImportGraphDef(Cxt->Graph, Cxt->Buffer, Cxt->GraphOpts, Cxt->Stat);
  if (unlikely(TF_GetCode(Cxt->Stat) != TF_OK)) {
    spdlog::error("[WasmEdge-Tensorflow] Cannot import graph from buffer: {}"sv,
                  TF_Message(Cxt->Stat));
    Env.deleteContext(NewID);
    return static_cast<uint32_t>(ErrNo::InvalidArgument);
  }

  // Create session.
  Cxt->SessionOpts = TF_NewSessionOptions();
  Cxt->Session = TF_NewSession(Cxt->Graph, Cxt->SessionOpts, Cxt->Stat);
  if (unlikely(TF_GetCode(Cxt->Stat) != TF_OK)) {
    spdlog::error("[WasmEdge-Tensorflow] Unable to create session: {}"sv,
                  TF_Message(Cxt->Stat));
    Env.deleteContext(NewID);
    return static_cast<uint32_t>(ErrNo::InvalidArgument);
  }

  *SessionId = NewID;
  return static_cast<uint32_t>(ErrNo::Success);
}

Expect<uint32_t> CreateSessionSavedModel::body(
    const Runtime::CallingFrame &Frame, uint32_t PathPtr, uint32_t PathLen,