-
Notifications
You must be signed in to change notification settings - Fork 3.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[REFACTOR][TYPE] Finish move all types to IR. (#4746)
* [REFACTOR][TYPE] Finish move all types to IR. - Move definition of Ref and TensorType to ir - Move type_functor.h to public header. - Rename RefType -> RelayRefType for clarity. * Add atol
- Loading branch information
Showing
56 changed files
with
486 additions
and
408 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,117 @@ | ||
| /* | ||
| * 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/ir/tensor_type.h | ||
| * \brief Polymorphic tensor types. | ||
| */ | ||
| #ifndef TVM_IR_TENSOR_TYPE_H_ | ||
| #define TVM_IR_TENSOR_TYPE_H_ | ||
|
|
||
| #include <tvm/ir/type.h> | ||
| #include <tvm/ir/expr.h> | ||
|
|
||
| namespace tvm { | ||
| /*! | ||
| * \brief Base of all Tensor types | ||
| * This container can hold TensorType or GenericTensorType. | ||
| * \sa BaseTensorType, TensorTypeNode | ||
| */ | ||
| class BaseTensorTypeNode : public TypeNode { | ||
| public: | ||
| static constexpr const char* _type_key = "relay.BaseTensorType"; | ||
| TVM_DECLARE_BASE_OBJECT_INFO(BaseTensorTypeNode, TypeNode); | ||
| }; | ||
|
|
||
| /*! | ||
| * \brief Managed reference to BaseTensorTypeNode. | ||
| * \sa BaseTensorTypeNode. | ||
| */ | ||
| class BaseTensorType : public Type { | ||
| public: | ||
| TVM_DEFINE_OBJECT_REF_METHODS(BaseTensorType, Type, BaseTensorTypeNode); | ||
| }; | ||
|
|
||
| /*! | ||
| * \brief This is the most commonly used type in relay. | ||
| * TensorType have a fixed dimension, data type. | ||
| * | ||
| * The elements of shape can be either IntImm(constant integer), | ||
| * or any symbolic integer expression. | ||
| * The symbolic integer allows generic shape inference in certain cases. | ||
| * \sa TensorType | ||
| */ | ||
| class TensorTypeNode : public BaseTensorTypeNode { | ||
| public: | ||
| /*! | ||
| * \brief The shape of the tensor, | ||
| * represented by PrimExpr(tvm::Expr). | ||
| */ | ||
| Array<PrimExpr> shape; | ||
| /*! \brief The content data type */ | ||
| DataType dtype; | ||
|
|
||
| void VisitAttrs(tvm::AttrVisitor* v) { | ||
| v->Visit("shape", &shape); | ||
| v->Visit("dtype", &dtype); | ||
| v->Visit("span", &span); | ||
| } | ||
|
|
||
| /*! \brief Return product of elements in the shape. | ||
| * \return (d1 * d_2 ... * d_n) if shape is (d_1, d_2, ..., d_n) and 1 if shape size is zero. | ||
| */ | ||
| TVM_DLL PrimExpr Size() const; | ||
|
|
||
| static constexpr const char* _type_key = "relay.TensorType"; | ||
| TVM_DECLARE_FINAL_OBJECT_INFO(TensorTypeNode, BaseTensorTypeNode); | ||
| }; | ||
|
|
||
| /*! | ||
| * \brief Managed reference to TensorTypeNode. | ||
| * \sa TensorTypeNode. | ||
| */ | ||
| class TensorType : public Type { | ||
| public: | ||
| /*! | ||
| * \brief Constructor. | ||
| * \param shape The shape of the tensor. | ||
| * \param dtype The runtime dtype of the tensor's elements. | ||
| */ | ||
| TVM_DLL TensorType(Array<PrimExpr> shape, DataType dtype); | ||
|
|
||
| /*! | ||
| * \brief Construct an scalar containing elements of dtype. | ||
| * \param dtype The runtime dtype of the tensor's elements. | ||
| * \return THe constructed type. | ||
| */ | ||
| TVM_DLL static TensorType Scalar(DataType dtype); | ||
|
|
||
| TVM_DEFINE_OBJECT_REF_METHODS(TensorType, Type, TensorTypeNode); | ||
| }; | ||
|
|
||
| // The following fields contains advanced typing | ||
| // Only keep the class name and reserved for future usage. | ||
| class GenericTensorType; | ||
| // stores a DataType. | ||
| class GenericDataType; | ||
| // stores a DataType. | ||
| class GenericShape; | ||
|
|
||
| } // namespace tvm | ||
| #endif // TVM_IR_TENSOR_TYPE_H_ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.