Files
pytorch/torch/csrc/jit/codegen/onednn/decompose_silu.h
2024-10-26 17:41:27 +00:00

10 lines
188 B
C++

#pragma once
#include <torch/csrc/jit/ir/ir.h>
namespace torch::jit::fuser::onednn {
void DecomposeSiluForLLGA(std::shared_ptr<Graph>& graph);
} // namespace torch::jit::fuser::onednn