/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

/*!
 * \file tvm/relay/qnn/transform.h
 *
 * This file implements a pass manager for QNN ops using Relay Pass manager.
 */
#ifndef TVM_RELAY_QNN_TRANSFORM_H_
#define TVM_RELAY_QNN_TRANSFORM_H_

#include <tvm/runtime/c_runtime_api.h>
#include <tvm/relay/transform.h>

namespace tvm {
namespace relay {

using relay::transform::Pass;

namespace qnn {
namespace transform {

/*!
 * \brief Legalizes a QNN expr. Contains specifically two types of Legalizations. First,
 * converts/Lowers an expression containing QNN ops to an expression containing only core Relay ops.
 * Each QNN op is lowered to a sequence of exisiting Relay ops. This is a target-independent pass.
 * One can register the lowering/transformation function for this op using FTVMQnnCanonicalize
 * attr_name for FTVMLegalize op attribute. Second, as opposed to Relay Legalize, this one legalizes
 * only QNN ops. One can register a transformation/legalization function for an op by using the
 * FTVMQnnLegalize attr_name for FTVMLegalize op attribute. The isolation of QNN and Relay Legalize
 * gives us separation of concerns, leading to a better software practice. The legalization can be
 * configured to happen per target.
 *
 * \return The pass.
 */
TVM_DLL Pass Legalize();

}  // namespace transform

}  // namespace qnn
}  // namespace relay
}  // namespace tvm

#endif  // TVM_RELAY_QNN_TRANSFORM_H_