(input_)
| 132 | return _CopyToModelParallelRegion.apply(input_) |
| 133 | |
| 134 | def reduce_from_model_parallel_region(input_): |
| 135 | return _ReduceFromModelParallelRegion.apply(input_) |
| 136 | |
| 137 | def scatter_to_model_parallel_region(input_): |
| 138 | return _ScatterToModelParallelRegion.apply(input_) |