1 #pragma once 2 3 #include <torch/csrc/jit/ir/ir.h> 4 5 namespace torch { 6 namespace jit { 7 namespace fuser { 8 namespace onednn { 9 10 void prepareFusionGroupAndGuardOutputs(Block* block); 11 12 } // namespace onednn 13 } // namespace fuser 14 } // namespace jit 15 } // namespace torch 16