pytorch/caffe2/python/test
Hector Yuen c8e789e06e add fake fp16 fusions to net transforms (#42927)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/42927

added fp16 fusion to net transforms
refactored the transforms as well as glow_transform to get out of opt/custom so that the OSS builds passed

Test Plan: added net runner tests for this

Reviewed By: yinghai

Differential Revision: D23080881

fbshipit-source-id: ee6451811fedfd07c6560c178229854bca29301f
2020-08-14 13:30:27 -07:00
..
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
blob_deallocation_test.py
do_op_test.py
executor_test.py Unify gpu_support variable in python tests (#16748) 2019-02-07 00:29:51 -08:00
executor_test_util.py [PyFI] Update hypothesis and switch from tp2 (#41645) 2020-08-08 12:13:04 -07:00
fakefp16_transform_test.py add fake fp16 fusions to net transforms (#42927) 2020-08-14 13:30:27 -07:00
gpu_context_test.py Correct #39759 for HIP. (#39801) 2020-06-12 10:34:28 -07:00
inference_lstm_op_test.py [PyFI] Update hypothesis and switch from tp2 (#41645) 2020-08-08 12:13:04 -07:00
python_protobuf_test.py Enforce import order to make protobuf cpp implementation in python work (#18560) 2019-04-03 13:17:08 -07:00