(input_, dim, kernel_size)
| 281 | |
| 282 | |
| 283 | def conv_scatter_to_context_parallel_region(input_, dim, kernel_size): |
| 284 | return _ConvolutionScatterToContextParallelRegion.apply(input_, dim, kernel_size) |
| 285 | |
| 286 | |
| 287 | def conv_gather_from_context_parallel_region(input_, dim, kernel_size): |