Skip to content

Commit

Permalink
Remove some dead todos
Browse files Browse the repository at this point in the history
  • Loading branch information
jroesch committed Oct 21, 2019
1 parent fa823b8 commit 4354e06
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions src/relay/op/memory/memory.cc
Original file line number Diff line number Diff line change
Expand Up @@ -91,11 +91,7 @@ TVM_ADD_FILELINE)
return {topi::identity(inputs[0])};
});

// TODO(@jroesch): re-export from sep memory namespace
TVM_REGISTER_API("relay.op.memory._make.alloc_tensor")
// The passing value in attrs and args doesn't seem super great.
// We should consider a better solution, type relation looking at
// args?
.set_body_typed<Expr(Expr, Expr, DataType, Array<IndexExpr> assert_shape)>(
[](Expr storage, tvm::relay::Expr shape, DataType dtype, Array<IndexExpr> assert_shape) {
auto attrs = make_node<AllocTensorAttrs>();
Expand Down Expand Up @@ -264,7 +260,6 @@ TVM_ADD_FILELINE)
return {topi::identity(inputs[0])};
});

// TODO(@jroesch): re-export from sep memory namespace
TVM_REGISTER_API("relay.op.memory._make.shape_func")
.set_body_typed<Expr(Expr, Expr, Expr, bool)>([](Expr func, Expr inputs, Expr outputs, bool dependent) {
static const Op& op = Op::Get("memory.shape_func");
Expand Down

0 comments on commit 4354e06

Please sign in to comment.