Passing "Any" as an argument to Relay ops

MarisaKirisame · April 10, 2020, 7:59pm

think of it like this: a LSTM is a fold (recursive visit) on list, and a TreeLSTM is a fold (recursive visit) on rose tree.

data Rose a = MakeRose a (List (Rose a))

I had implemented treelstm a long time ago for relay benchmark, which is at here.

gist.github.com

https://gist.github.com/MarisaKirisame/d37ff587cb02d76e7e56125123f25310

network.py

import numpy as np
import tvm
from tvm import relay
from tvm.relay import op
from tvm.relay import create_executor, Module
from tvm.relay.backend.interpreter import TensorValue
from tvm.relay.prelude import Prelude
from tvm.relay.testing import add_nat_definitions
import aot
import collections

This file has been truncated. show original

tlstm.py

from .network import Network
from tvm import relay
from tvm.relay import op, var, Var, Function, Clause, PatternConstructor, PatternVar, Match
from tvm.relay import TupleGetItem, Tuple, TensorType, TupleType

class Linear(Network):
    def build_impl(self, input_size, output_size, dtype="float32"):
        x = self.input(var("linear_input", shape=(1, input_size), dtype=dtype))
        w = self.weight(var("linear_weight", shape=(output_size, input_size), dtype=dtype))
        b = self.weight(var("linear_bias", shape=(output_size,), dtype=dtype))

This file has been truncated. show original