search

cornell-zhang / heterocl

HeteroCL: A Multi-Paradigm Programming Infrastructure for Software-Defined Heterogeneous Computing
https://cornell-zhang.github.io/heterocl/
Apache License 2.0
322 stars 92 forks source link

[API][v0.3] Connect HeteroCL with Keras via Relay #142

Closed PBC48 closed 4 years ago

PBC48 commented 4 years ago

I have implemented a frontend for HeteroCL in the hlib directory. In the hlib directory there are two directories: frontend and op. Frontend contains the parser/transpiler and also a test directory that contains the tests that verify both the parser/transpiler and the backend functions in the op directory. Inside the op directory, there are three files: math.py, nn.py, and op.py. All of these files were formatted to meet the PEP8 standards.

I have not extended CircleCI to the tests, but I can do so in another pull

PBC48 commented 4 years ago

Good question. We don't

seanlatias commented 4 years ago

Could you tell me which files I should modify when I remove the two return objects?

PBC48 commented 4 years ago

wait, actually it does need those two other outputs to be consistent with relay API

PBC48 commented 4 years ago

https://docs.tvm.ai/langref/relay_op.html?highlight=batch_norm#tvm.relay.nn.batch_norm