mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
b07ea91c4c
Follows #131034 Pull Request resolved: https://github.com/pytorch/pytorch/pull/131735 Approved by: https://github.com/ezyang
20 lines
595 B
C++
20 lines
595 B
C++
#pragma once
|
|
|
|
#include <caffe2/serialize/versions.h>
|
|
#include <torch/csrc/Export.h>
|
|
#include <torch/csrc/jit/api/module.h>
|
|
|
|
#include <cstdint>
|
|
|
|
namespace torch::jit {
|
|
// Maps the given symbol into an implementation of its behavior at the
|
|
// given version.
|
|
// See note [Versioned Symbols]
|
|
TORCH_API Symbol
|
|
get_symbol_for_version(const Symbol name, const uint64_t version);
|
|
|
|
// Maps the given kind to the minimum version that supports it.
|
|
// See note [Dynamic Versions and torch.jit.save vs. torch.save]
|
|
TORCH_API uint64_t get_min_version_for_kind(const NodeKind& kind);
|
|
} // namespace torch::jit
|