()
| 38 | |
| 39 | |
| 40 | def GetArgumentParser(): |
| 41 | parser = argparse.ArgumentParser(description="Caffe2 optimization") |
| 42 | parser.add_argument("--init_net", |
| 43 | type=argparse.FileType('rb'), |
| 44 | help="init net") |
| 45 | parser.add_argument("--pred_net", |
| 46 | type=argparse.FileType('rb'), |
| 47 | help="predict net") |
| 48 | parser.add_argument("--verify_input", |
| 49 | type=argparse.FileType('r'), |
| 50 | help="input dims for verification") |
| 51 | parser.add_argument("--fuse_bn", default=False, action='store_true') |
| 52 | parser.add_argument("--fuse_mul_add", default=False, action='store_true') |
| 53 | parser.add_argument("--fuse_conv_relu", default=False, action='store_true') |
| 54 | return parser |
| 55 | |
| 56 | |
| 57 | def fuse_first_bn(net, params, removed_tensors): |
no outgoing calls
no test coverage detected
searching dependent graphs…