pytorch/caffe2/python/predictor
Bangsheng Tang 8f854fb9e2 [1/n][multi-tower] add partition info in predictor construction (#34175)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34175

to incorporate PartitionInfo added in D20015493

Test Plan: unit tests

Reviewed By: yinghai

Differential Revision: D20133759

fbshipit-source-id: 130db2d80bca3c05a7ec91292159f857046718e0
2020-03-13 09:23:39 -07:00
..
__init__.py
mobile_exporter.py
mobile_exporter_test.py
predictor_exporter.py
predictor_exporter_test.py fix fc fp16 quantization (#29469) 2019-11-18 11:26:49 -08:00
predictor_py_utils.py [1/n][multi-tower] add partition info in predictor construction (#34175) 2020-03-13 09:23:39 -07:00
predictor_test.py
serde.py