1/*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19/*!
20 * \file operation_inline.h
21 */
22#ifndef TVM_TE_SCHEDULE_OPERATION_INLINE_H_
23#define TVM_TE_SCHEDULE_OPERATION_INLINE_H_
24
25#include <tvm/te/operation.h>
26#include <tvm/te/tensor.h>
27#include <tvm/tir/expr.h>
28#include <tvm/tir/stmt.h>
29
30namespace tvm {
31namespace te {
32
33/*!
34 * \brief inline all calls of f in stmt.
35 *
36 * \param stmt The statement to apply inline optimization.
37 * \param op The op to be inlined.
38 * \param args The arguments variable of the function.
39 * \param body The definition body of the function.
40 * \return The result stmt
41 *
42 * \note All the passes in this file uses SSA form and outputs SSA form.
43 */
44Stmt Inline(Stmt stmt, Operation op, Array<Var> args, PrimExpr body);
45
46} // namespace te
47} // namespace tvm
48#endif // TVM_TE_SCHEDULE_OPERATION_INLINE_H_
49