Refactor type inference

pfnet-research / chainer-compiler

Experimental toolchain to compile and run Chainer models

MIT License

112 stars 23 forks source link

Closed momohatt closed 4 years ago

momohatt commented 4 years ago

This PR adds minor refinements to the type inference engine, including:

Adding type signature for torch.nn.Embedding (which I implemented in attempt to typecheck SNLI example but gave up in the middle because the program creates user-defined object in forward)
Add a function to compute the type of x.shape and x.size (where x is a tensor), which used to be embedded in the inference engine

shinh commented 4 years ago

/test

pfn-ci-bot commented 4 years ago

Successfully created a job for commit b91c380: