mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
b0fdca8855
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68358 Test Plan: Imported from OSS Reviewed By: albanD Differential Revision: D33433730 Pulled By: tugsbayasgalan fbshipit-source-id: 202c58365bae13195d3545cefcb0da9162b02151
24 lines
649 B
C++
24 lines
649 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 {
|
|
namespace jit {
|
|
#if !ENABLE_UPGRADERS
|
|
// 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);
|
|
#endif
|
|
} // namespace jit
|
|
} // namespace torch
|