(model)
| 43 | |
| 44 | |
| 45 | def get_all_params(model): |
| 46 | params = [] |
| 47 | for node in model.walk(): |
| 48 | for name in node.param_names: |
| 49 | params.append(node.get_param(name).ravel()) |
| 50 | return node.ops.xp.concatenate(params) |
| 51 | |
| 52 | |
| 53 | def get_docs(): |
no test coverage detected
searching dependent graphs…