(model_fn, features, devices)
| 90 | |
| 91 | |
| 92 | def parallel_model(model_fn, features, devices): |
| 93 | if len(devices) == 1: |
| 94 | return [model_fn(features)] |
| 95 | |
| 96 | features = shard_features(features, devices) |
| 97 | |
| 98 | outputs = data_parallelism(devices, model_fn, features) |
| 99 | return outputs |
nothing calls this directly
no test coverage detected