A function that reproduces Caffe's inclusion and exclusion rule.
(net_state, layer)
| 37 | |
| 38 | |
| 39 | def _ShouldInclude(net_state, layer): |
| 40 | """A function that reproduces Caffe's inclusion and exclusion rule.""" |
| 41 | ret = (len(layer.include) == 0) |
| 42 | # check exclude rules: if any exclusion is met, we shouldn't include. |
| 43 | ret &= not any([_StateMeetsRule(net_state, rule) for rule in layer.exclude]) |
| 44 | if len(layer.include): |
| 45 | # check include rules: if any inclusion is met, we should include. |
| 46 | ret |= any([_StateMeetsRule(net_state, rule) for rule in layer.include]) |
| 47 | return ret |
| 48 | |
| 49 | |
| 50 | def _GetLegacyDims(net, net_params, dummy_input, legacy_pad_ops): |
no test coverage detected
searching dependent graphs…