29 #ifndef MXNET_EXPR_OPERATOR_H_ 30 #define MXNET_EXPR_OPERATOR_H_ 36 template<
typename ValueType>
38 if (t.
is_int())
return IntImm(t, static_cast<int64_t>(value));
41 LOG(FATAL) <<
"cannot make const for type " << t;
46 template<
typename ValueType>
51 LOG(FATAL) <<
"MXNetDataType::lanes() != 1 is not supported ";
58 #endif // MXNET_EXPR_OPERATOR_H_ int lanes() const
Definition: data_type.h:82
Managed reference class to FloatImmNode.
Definition: expr.h:197
PrimExpr make_const(MXNetDataType t, ValueType value)
Definition: expr_operator.h:47
namespace of mxnet
Definition: api_registry.h:33
bool is_float() const
Definition: data_type.h:94
Managed reference class to IntImmNode.
Definition: expr.h:152
Reference to PrimExprNode.
Definition: expr.h:101
PrimExpr MakeConstScalar(MXNetDataType t, ValueType value)
Definition: expr_operator.h:37
Runtime primitive data type.
Definition: data_type.h:41
Base expr nodes in MXNet.
bool is_int() const
Definition: data_type.h:98