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 9 namespace torch::jit { 10 11 TORCH_API void unprofileGraphInputs(const std::shared_ptr<Graph>& graph); 12 TORCH_API void unprofileBlock(Block* start_block); 13 // Unprofiles all the node outputs in a block. 14 15 TORCH_API void ClearProfilingInformation(const std::shared_ptr<Graph>& graph); 16 17 } // namespace torch::jit 18