Simplify some caffe ops Pattern 1: combine multi-slices to one Pattern 2: combine multi-eltwise back to one.
(caffe_net: ppl_caffe_pb2.NetParameter)
| 97 | return new_net |
| 98 | |
| 99 | def optimize_for_export(caffe_net: ppl_caffe_pb2.NetParameter) -> ppl_caffe_pb2.NetParameter: |
| 100 | """Simplify some caffe ops Pattern 1: combine multi-slices to one Pattern |
| 101 | 2: combine multi-eltwise back to one.""" |
| 102 | slice_combine(caffe_net) |
| 103 | eltwise_combine(caffe_net) |
| 104 | return caffe_net |
| 105 | |
| 106 | |
| 107 | def slice_combine(caffe_net: ppl_caffe_pb2.NetParameter): |
no test coverage detected