Function setInput

plugins/wasi_nn/wasinn_piper.cpp:291–336

}

Expect<WASINN::ErrNo> setInput(WASINN::WasiNNEnvironment &Env,
                               uint32_t ContextId, uint32_t Index,
                               const TensorData &Tensor) noexcept {
  if (Index != 0) {
    spdlog::error("[WASI-NN] Piper backend: Input index must be 0."sv);
    return WASINN::ErrNo::InvalidArgument;
  }
  if (Tensor.Dimension.size() != 1) {
    spdlog::error(
        "[WASI-NN] Piper backend: Input tensor dimension must be 1D."sv);
    return WASINN::ErrNo::InvalidArgument;
  }

  auto &CxtRef = Env.NNContext[ContextId].get<Context>();
  auto &GraphRef = Env.NNGraph[CxtRef.GraphId].get<Graph>();

  CxtRef.Line =
      std::string(reinterpret_cast<const char *>(Tensor.Tensor.data()),
                  Tensor.Tensor.size());

  if (GraphRef.Config->JsonInput) {
    simdjson::dom::parser Parser;
    simdjson::dom::element Doc;
    simdjson::padded_string const PaddedInput(CxtRef.Line.value());

    if (Parser.parse(PaddedInput).get(Doc) != simdjson::SUCCESS) {
      spdlog::error("[WASI-NN] Piper backend: Failed to parse JSON input."sv);
      return WASINN::ErrNo::InvalidArgument;
    }

    simdjson::dom::object JsonObj;
    if (Doc.get(JsonObj) != simdjson::SUCCESS) {
      spdlog::error("[WASI-NN] Piper backend: JSON input is not an object."sv);
      return WASINN::ErrNo::InvalidArgument;
    }

    SynthesisConfig NewConfig;
    if (auto Err = parseSynthesisConfig(NewConfig, JsonObj);
        Err != WASINN::ErrNo::Success) {
      return Err;
    }
    CxtRef.JsonInputSynthesisConfig =
        std::make_unique<std::optional<SynthesisConfig>>(NewConfig);
  }
  return WASINN::ErrNo::Success;
}

Expect<WASINN::ErrNo> getOutput(WASINN::WasiNNEnvironment &Env,
                                uint32_t ContextId, uint32_t Index,