Go to the documentation of this file.
29 #ifndef MXNET_EXPR_OPERATOR_H_
30 #define MXNET_EXPR_OPERATOR_H_
36 template <
typename ValueType>
39 return IntImm(t,
static_cast<int64_t
>(value));
41 return FloatImm(t,
static_cast<double>(value));
43 LOG(FATAL) <<
"cannot make const for type " << t;
47 template <
typename ValueType>
52 LOG(FATAL) <<
"MXNetDataType::lanes() != 1 is not supported ";
59 #endif // MXNET_EXPR_OPERATOR_H_
namespace of mxnet
Definition: api_registry.h:33
Base expr nodes in MXNet.
int lanes() const
Definition: data_type.h:80
PrimExpr make_const(MXNetDataType t, ValueType value)
Definition: expr_operator.h:48
Managed reference class to FloatImmNode.
Definition: expr.h:197
bool is_float() const
Definition: data_type.h:92
Reference to PrimExprNode.
Definition: expr.h:101
bool is_int() const
Definition: data_type.h:96
Runtime primitive data type.
Definition: data_type.h:40
PrimExpr MakeConstScalar(MXNetDataType t, ValueType value)
Definition: expr_operator.h:37
Managed reference class to IntImmNode.
Definition: expr.h:152