1#pragma once
2
3#include <torch/csrc/jit/ir/ir.h>
4
5namespace torch {
6namespace jit {
7
8TORCH_API void BatchMM(std::shared_ptr<Graph>& graph);
9
10}
11} // namespace torch
12