chromium/third_party/mediapipe/src/mediapipe/util/tflite/tflite_signature_reader.cc

#include "mediapipe/util/tflite/tflite_signature_reader.h"

#include <algorithm>
#include <cstdint>
#include <map>
#include <optional>
#include <string>
#include <utility>
#include <vector>

#include "absl/container/flat_hash_map.h"
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/str_join.h"
#include "mediapipe/framework/port/ret_check.h"
#include "mediapipe/framework/port/status_macros.h"
#include "tensorflow/lite/interpreter.h"

namespace mediapipe {

namespace {

// Flips the key-value pairs in a map.
absl::flat_hash_map<uint32_t, std::string> FlipKVInMap(
    const std::map<std::string, uint32_t>& map) { … }

}  // namespace

absl::StatusOr<SignatureInputOutputTensorNames>
TfLiteSignatureReader::GetInputOutputTensorNamesFromTfliteSignature(
    const tflite::Interpreter& interpreter, const std::string* signature_key) { … }

absl::StatusOr<
    absl::flat_hash_map<SignatureName, SignatureInputOutputTensorNames>>
TfLiteSignatureReader::GetInputOutputTensorNamesFromAllTfliteSignatures(
    const tflite::Interpreter& interpreter) { … }
}  // namespace mediapipe