projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: d4f6bef) | patch
RNN operators should inherit step_net device_options (#16086)
authorNikita Shulga <nikita.shulga@oculus.com>
Fri, 18 Jan 2019 19:33:40 +0000 (11:33 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 18 Jan 2019 19:36:38 +0000 (11:36 -0800)
commitb69c05dbd61901882a58df112e20491e8c2b8794
tree4694e632374b3e99784e75b172b3451ac5064eaatree | snapshot
parentd4f6befc93938e5462b186f76405b544e50fab60commit | diff
RNN operators should inherit step_net device_options (#16086)

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

[caffe2] RNN operators should inherit step_net device_options
According to NetDef documentaiton, if network has a specific device option it applies to all network operators that do not explicitly specifiy it.
But this does not seem to be the case for RecurrentNetwork operators

Reviewed By: orionr

Differential Revision: D13699552

fbshipit-source-id: 14529bc9504e3b02f763e3c2429be21e46f82b68
caffe2/operators/rnn/recurrent_network_executor.h diff | blob | history
Domain: Machine Learning / ML Framework;