Lu Fang
52460a0b30
Add outputs_info as parameter in run_node ( #2161 )
2018-03-09 10:44:51 -08:00
anderspapitto
c0304c83b1
Copy some outputs in order to decouple storage ( #2105 )
...
so that mutating one of them does not mutate the others
2018-03-01 13:25:31 -08:00
anderspapitto
749a17661c
Introduce padding op to mimic pytorch semantics in ONNX export ( #2069 )
...
In pytorch, after pad_packed_sequence, the "extra" elements (after the
ends of the sequences) are reset. In the equivalent Caffe2 graph
exported via ONNX, they contained some leftover values, which caused
tests to fail. Probably no one depends on these values, but just in
case, set them to zero to mimic pytorch semantics.
2018-02-28 15:44:54 -08:00
anderspapitto
76304300a8
Transpose shape inference ( #2057 )
...
* fix variable name
* enhance shape inference to handle transpose
in the case arising from pack_padded(..., batch_first=True)
2018-02-27 11:51:10 -08:00
anderspapitto
ec547ce640
RNN ONNX export: concat hidden/cell states on the right axis ( #2055 )
...
Test Plan: existing tests in onnx-fb-universe catch this, modulo a bug
in the tests which I am fixing in a separate diff
2018-02-26 11:04:04 -08:00
anderspapitto
38f18c1daa
add third output in onnx -> caffe2 lstm conversion ( #2011 )
2018-02-22 17:43:33 -08:00
Yinghai Lu
cc7e61c88d
Move onnx-caffe2 inside caffe2 ( #1921 )
...
* Move onnx-caffe2 inside caffe2
* Update to the lastest onnx-caffe2 and update jenkins env
* Rename onnx_caffe2 to onnx
* Add __init__.py to caffe2/python/onnx
* Change CI check variable to JENKINS_URL
* Cherrypick recent onnx-caffe2 update
2018-02-20 13:56:52 -08:00
