MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / conv_pass_from_last_rank

Function conv_pass_from_last_rank

sat/sgm/modules/cp_enc_dec.py:291–292  ·  view source on GitHub ↗
(input_, dim, kernel_size)

Source from the content-addressed store, hash-verified

289
290
291def conv_pass_from_last_rank(input_, dim, kernel_size):
292 return _ConvolutionPassFromPreviousRank.apply(input_, dim, kernel_size)
293
294
295class ContextParallelCausalConv3d(nn.Module):

Callers 1

forwardMethod · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected