/* * 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. */ /*! * \brief Placeholder op. * \file placeholder_op.cc */ #include <tvm/runtime/registry.h> #include <tvm/te/operation.h> namespace tvm { namespace te { // PlaceholderOpNode TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable) .set_dispatch<PlaceholderOpNode>([](const ObjectRef& node, ReprPrinter* p) { auto* op = static_cast<const PlaceholderOpNode*>(node.get()); p->stream << "placeholder(" << op->name << ", " << op << ")"; }); TVM_REGISTER_NODE_TYPE(PlaceholderOpNode); int PlaceholderOpNode::num_outputs() const { return 1; } Array<IterVar> PlaceholderOpNode::root_iter_vars() const { return {}; } DataType PlaceholderOpNode::output_dtype(size_t i) const { CHECK_EQ(i, 0U); return dtype; } Array<PrimExpr> PlaceholderOpNode::output_shape(size_t i) const { CHECK_EQ(i, 0U); return shape; } Operation PlaceholderOpNode::make(std::string name, Array<PrimExpr> shape, DataType dtype) { auto n = make_object<PlaceholderOpNode>(); n->name = name; n->shape = shape; n->dtype = dtype; return Operation(n); } Tensor placeholder(Array<PrimExpr> shape, DataType dtype, std::string name) { return PlaceholderOpNode::make(name, shape, dtype).output(0); } TVM_REGISTER_GLOBAL("te.Placeholder") .set_body_typed([](Array<PrimExpr> shape, DataType dtype, std::string name) { return placeholder(shape, dtype, name); }); Array<Tensor> PlaceholderOpNode::InputTensors() const { return {}; } Operation PlaceholderOpNode::ReplaceInputs( const Operation& self, const std::unordered_map<Tensor, Tensor>& rmap) const { return self; } void PlaceholderOpNode::PropBoundToInputs( const Operation& self, arith::Analyzer* analyzer, const std::unordered_map<const VarNode*, IntSet>& dom_map, std::unordered_map<Tensor, TensorDom>* out_dom_map) const { } void PlaceholderOpNode::GatherBound( const Operation& self, const std::unordered_map<Tensor, TensorDom>& tensor_dom, std::unordered_map<IterVar, Range>* out_dom_map) const { } Stmt PlaceholderOpNode::BuildRealize( const Stage& stage, const std::unordered_map<IterVar, Range>& realize_map, const Stmt& body) const { return body; } Stmt PlaceholderOpNode::BuildProvide( const Stage& stage, const std::unordered_map<IterVar, Range>& dom_map, bool debug_keep_trivial_loop) const { return Stmt(); } } // namespace te } // namespace tvm