pytorch/caffe2/python/test
Hector Yuen a4b763bc2c add net transforms for fusion (#42763)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/42763

add the fp16 fusions as net transforms:
-layernorm fused with mul+add
-swish int8

Test Plan: added unit test, ran flows

Reviewed By: yinghai

Differential Revision: D23002043

fbshipit-source-id: f0b13d51d68c240b05d2a237a7fb8273e996328b
2020-08-10 20:16:14 -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
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 net transforms for fusion (#42763) 2020-08-10 20:16:14 -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