parameters: (self, op)
docstring: Visit op node
docstring_summary: Visit op node
docstring_tokens: Visit op node
function: def visit(self, op):
          """Visit op node"""
          prim = PrimLib.get_prim(op)
          if prim.iter_type in (PrimLib.ELEMWISE, PrimLib.BROADCAST, PrimLib.REDUCE):
              out_dim = len(op.output.shape)
              align_dim = out_dim
              for t in op.inputs:
                  if len(t.shape) > align_dim:
                      align_dim = len(t.shape)
              if align_dim > out_dim:
                  op.output.shape = [1] * (align_dim - out_dim) + op.output.shape
function_tokens: def visit ( self , op ) : prim = PrimLib . get_prim ( op ) if prim . iter_type in ( PrimLib . ELEMWISE , PrimLib . BROADCAST , PrimLib . REDUCE ) : out_dim = len ( op . output . shape ) align_dim = out_dim for t in op . inputs : if len ( t . shape ) > align_dim : align_dim = len ( t . shape ) if align_dim > out_dim : op . output . shape = [ 1 ] * ( align_dim - out_dim ) + op . output . shape