Program Listing for File NPUSerialization.h

↰ Return to documentation for file (csrc/backend/NPUSerialization.h)

#pragma once

#include <ATen/Tensor.h>
#include <string>
#include <unordered_map>

namespace c10::backend {
// Serialize npu-related information, mainly related to private formats
void npu_info_serialization(
    const at::Tensor& t,
    std::unordered_map<std::string, bool>& mate_map);
// Deserialize npu-related information, mainly related to private formats
void npu_info_deserialization(
    const at::Tensor& t,
    std::unordered_map<std::string, bool>& mate_map);
} // namespace c10::backend