def set_p2p_tensor_shapes()

in dualpipe/comm.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

def set_p2p_tensor_shapes(shapes: List[Tuple[int]]):
    global TENSOR_SHAPES
    TENSOR_SHAPES = shapes