mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
149f5ffa36
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71185 Test Plan: Imported from OSS Reviewed By: gmagogsfm Differential Revision: D33539191 Pulled By: tugsbayasgalan fbshipit-source-id: 721093793574663d56a8080c6a488024620266a1
22 lines
620 B
C++
22 lines
620 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 {
|
|
// 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 jit
|
|
} // namespace torch
|