projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 66b2680) | patch
caffe2 - make DataRandomFiller usable in unit tests (#15027)
authorDuc Ngo <duc@fb.com>
Fri, 14 Dec 2018 04:43:00 +0000 (20:43 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 14 Dec 2018 04:45:52 +0000 (20:45 -0800)
commit331c4b5b4d600049fbd5a6503c9cf5f1009214c5
tree112ea507c25696f83094c1d103e07a48b6f1c146tree | snapshot
parent66b26806fc6e9a0a2db24d31cfff4f88c1d51ae6commit | diff
caffe2 - make DataRandomFiller usable in unit tests (#15027)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15027

- Make DataRandomFiller able to accept input_dims and input_types for only non intermediate inputs. Add a helper to fill input directly to a workspace

Reviewed By: highker

Differential Revision: D13408345

fbshipit-source-id: 5fc54d33da12e3f0a200e79380d4c695b0339b17
caffe2/predictor/emulator/data_filler.cc diff | blob | history
caffe2/predictor/emulator/data_filler.h diff | blob | history
Domain: Machine Learning / ML Framework;