pytorch/caffe2/python/models
James Reed ab62a92dab De-dup beam search state reshape shape blob
Summary:
T22119644 showed that there is a potential illegal memory access in beam search with attention. Upon further inspection, we can see that there are multiple ops that write to the same old shape blob:

  {"output0": "model0/attention_decoder/attention_weighted_encoder_context_reshaped", "output1": "state_old_shape_before_choosing_per_hypo", "input0": "model0/attention_decoder/attention_weighted_encoder_context" }},
  {"output0": "model0/attention_decoder/hidden_t_external_reshaped", "output1": "state_old_shape_before_choosing_per_hypo", "input0": "model0/attention_decoder/hidden_t_external" }},
  {"output0": "model0/decoder/layer0/cell_t_reshaped", "output1": "state_old_shape_before_choosing_per_hypo", "input0": "model0/decoder/layer0/cell_t" }},

This diff de-dupes these outputs

Reviewed By: akyrola

Differential Revision: D5899103

fbshipit-source-id: 8b6f3f113e764dfeb9262f6c442e1124559cd2d8
2017-09-23 23:19:44 -07:00
..
seq2seq De-dup beam search state reshape shape blob 2017-09-23 23:19:44 -07:00
__sym_init__.py exec_net --> predict_net 2017-03-23 16:31:49 -07:00
download.py Add an argument for suppressing download progress 2017-09-17 20:15:51 -07:00
resnet_test.py fast simple-net memonger for C++ 2017-07-06 15:17:07 -07:00
resnet.py Fix a few typos and grammars in comment 2017-06-14 18:22:39 -07:00