FM / test

/usr/local/lib64/python3.6/site-packages/caffe2/python/test/

/usr/local/lib64/python3.6/site-packages/caffe2/python/test

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
blob_deallocation_test.py	632	0644	edit dl rm
do_op_test.py	2055	0644	edit dl rm
executor_test.py	3039	0644	edit dl rm
executor_test_util.py	8058	0644	edit dl rm
fakefp16_transform_test.py	723	0644	edit dl rm
gpu_context_test.py	1121	0644	edit dl rm
inference_lstm_op_test.py	2190	0644	edit dl rm
python_protobuf_test.py	624	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/test/fakefp16_transform_test.py (723B)

import unittest from caffe2.python.fakefp16_transform_lib import fakeFp16FuseOps from caffe2.python import core class Transformer(unittest.TestCase): def test_fuse(self): net_swish = core.Net("test_swish") net_swish_init = core.Net("test_swish_init") deq = core.CreateOperator("Int8DequantizeNNPI", ["Xq"], ["X"]) swish = core.CreateOperator("SwishFakeFp16NNPI", ["X"], ["Y"]) quant = core.CreateOperator("Int8QuantizeNNPI", ["Y"], ["Y_q"]) net_swish.Proto().op.extend( [ deq, swish, quant ] ) print(net_swish.Proto()) out_net = fakeFp16FuseOps(net_swish.Proto()) assert(len(out_net.op) == 1)