1. # file : src/relay/pass/pattern_util.h
    2. template <typename T>
    3. T GetScalarFromConstant(Expr expr) {
    4.   const auto* n = expr.as<ConstantNode>();
    5.   CHECK(n->is_scalar());
    6.   return static_cast<T*>(n->data->data)[0];
    7. }
    9. /*!
    10.    * \brief Construct an int type.
    11.    * \param bits The number of bits in the type.
    12.    * \param lanes The number of lanes.
    13.    * \return The constructed data type.
    14.    */
    15.   static DataType Int(int bits, int lanes = 1) {
    16.     return DataType(kDLInt, bits, lanes);
    17.   }
    19.   /*!
    20.    * \brief Construct an uint type.
    21.    * \param bits The number of bits in the type.
    22.    * \param lanes The number of lanes
    23.    * \return The constructed data type.
    24.    */
    25.   static DataType UInt(int bits, int lanes = 1) {
    26.     return DataType(kDLUInt, bits, lanes);
    27.   }
    29.   /*!
    30.    * \brief Construct an uint type.
    31.    * \param bits The number of bits in the type.
    32.    * \param lanes The number of lanes
    33.    * \return The constructed data type.
    34.    */
    35.   static DataType Float(int bits, int lanes = 1) {
    36.     return DataType(kDLFloat, bits, lanes);
    37.   }
    39.   /*!
    40.    * \brief Construct a bool type.
    41.    * \param lanes The number of lanes
    42.    * \return The constructed data type.
    43.    */
    44.   static DataType Bool(int lanes = 1) {
    45.     return DataType::UInt(1, lanes);
    46.   }