pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

Ievgen Soboliev 1632f053e5 implement user-only metadata for input_record

Summary:
We want to implement request only net and to do this we decided to split the work into two parts. The first part will propagate required metadata and the second part will cut the nets properly.
This diff is to propagate request_only metadata across the layers.

A few notes about implementation:
  - Each layer contains a field request_only which can be set based on the input_record. If all the scalars from the input_record are marked request_only we mark a layer as request_only;
  - Sparse-To-Dense layer sets request_only metadata;
  - SigridTransformation and SparseLookup layers propagate request_only status;
  - As for now we join request_only and other sparse features together in input_record, but ideally we may want to separate this, because request_only should be served separately;

Reviewed By: xianjiec

Differential Revision: D4259505

fbshipit-source-id: db8a30ef92cba84f1a843981b9dde3a8b9633608

2016-12-15 12:01:29 -08:00

__init__.py

fbsync. TODO: check if build files need update.

2016-11-15 00:00:46 -08:00

batch_lr_loss.py

fbsync at f5a877

2016-11-18 15:41:06 -08:00

concat.py

fbsync at f5a877

2016-11-18 15:41:06 -08:00

dot_product.py

implement sparse nn using layers