1 #pragma once 2 3 #include <torch/csrc/jit/ir/ir.h> 4 5 #include <memory> 6 7 namespace torch::jit { 8 // see .cpp for docs 9 TORCH_API void RemoveInplaceOps(const std::shared_ptr<Graph>& graph); 10 11 TORCH_API void ImplicitCastForBinaryInplaceOps(Block* block); 12 } // namespace torch::jit 13