From 5f2fc0fd53a6ff244e1ae1144fc55a5be3220a64 Mon Sep 17 00:00:00 2001 From: Haichen Shen Date: Mon, 18 Feb 2019 16:08:24 -0800 Subject: [PATCH] Fix bug --- include/tvm/relay/attrs/transform.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/tvm/relay/attrs/transform.h b/include/tvm/relay/attrs/transform.h index a3d87f427c6fb..18bffff92e1e4 100644 --- a/include/tvm/relay/attrs/transform.h +++ b/include/tvm/relay/attrs/transform.h @@ -103,11 +103,11 @@ struct ArangeAttrs : public tvm::AttrsNode { DataType dtype; TVM_DECLARE_ATTRS(ArangeAttrs, "relay.attrs.ArangeAttrs") { - TVM_ATTR_FIELD(start).set_default(make_const(Int(32), 0)) + TVM_ATTR_FIELD(start).set_default(make_const(Float(32), 0)) .describe("Start of interval. The interval includes this value."); TVM_ATTR_FIELD(stop) .describe("Stop of interval. The interval does not include this value."); - TVM_ATTR_FIELD(start).set_default(make_const(Int(32), 1)) + TVM_ATTR_FIELD(step).set_default(make_const(Float(32), 1)) .describe("Spacing between values."); TVM_ATTR_FIELD(dtype).set_default(NullValue()) .describe("Target data type.");