| ## @package control_ops | |
| # Module caffe2.python.helpers.control_ops | |
| from caffe2.python.control_ops_util import add_if_op, add_while_op | |
| def cond(model, cond_blob, external_blobs, then_model, else_model=None): | |
| """Condition""" | |
| add_if_op( | |
| model.net, | |
| cond_blob, | |
| external_blobs, | |
| then_model.net, | |
| else_model.net if else_model else None) | |
| def loop(model, cond_blob, external_blobs, loop_model, cond_model=None): | |
| """Loop""" | |
| add_while_op( | |
| model.net, | |
| cond_blob, | |
| external_blobs, | |
| loop_model.net, | |
| cond_model.net if cond_model else None) | |