28 lines
711 B
C++
28 lines
711 B
C++
#pragma once
|
|
|
|
#include <torch/nativert/graph/Graph.h>
|
|
|
|
#include <torch/csrc/utils/generated_serialization_types.h>
|
|
|
|
namespace torch::nativert {
|
|
/**
|
|
* This file contains serialization utilities for Graph.
|
|
*
|
|
* There are two serialized representations we care about:
|
|
* - Json: stable but hard to work with, not really human readable
|
|
* - Debug format: human-readable, not stable.
|
|
*/
|
|
|
|
// Json -> Graph
|
|
std::unique_ptr<Graph> jsonToGraph(
|
|
const torch::_export::GraphModule& jsonGraph,
|
|
bool loadNodeMetadata = true);
|
|
|
|
bool isSymbolic(const torch::_export::Argument& arg);
|
|
|
|
Constant constantToValue(
|
|
const torch::_export::Argument& jsonArg,
|
|
bool loadNodeMetadata);
|
|
|
|
} // namespace torch::nativert
|