1#pragma once
2
3#include <ATen/ATen.h>
4#include <ATen/core/ivalue.h>
5#include <ATen/core/jit_type.h>
6#include <torch/csrc/Export.h>
7#include <torch/csrc/jit/ir/ir.h>
8
9namespace torch {
10namespace jit {
11
12TORCH_API void unprofileGraphInputs(const std::shared_ptr<Graph>& graph);
13TORCH_API void unprofileBlock(Block* start_block);
14// Unprofiles all the node outputs in a block.
15
16TORCH_API void ClearProfilingInformation(const std::shared_ptr<Graph>& graph);
17
18} // namespace jit
19} // namespace torch
20