nnvm/tvm/topi/python/topi/x86/injective.py
def schedule_injective(outs):
    """X86 schedule for injective op.

    Parameters
    ----------
    outs: Array of Tensor
          The computation graph description of injective in the format
          of an array of tensors.

    Returns
    -------
    sch: Schedule
        The computation schedule for the op.
    """
    outs = [outs] if isinstance(outs, tvm.tensor.Tensor) else outs
    x = outs[0]
    s = tvm.create_schedule([x.op for x in outs])
    tvm.schedule.AutoInlineInjective(s)
    if len(s[x].op.axis) == 4:
        n, c, _, _ = s[x].op.axis
        fused = s[x].fuse(n, c) # for nhwc layout, fuse n and h
        s[x].parallel(fused)
    else:
        s[x].parallel(s[x].op.axis[0])
    return s
X86 schedule for injective op.

    Parameters
    ----------
    outs: Array of Tensor
          The computation graph description of injective in the format
          of an array of tensors.

    Returns
    -------
    sch: Schedule
        The computation schedule for the op.
    
X86 schedule for injective op.
Parameters

Array of Tensor
The computation graph description of injective in the format
of an array of tensors.

Returns

Schedule
The computation schedule for the op.
X86 schedule for injective op . Parameters Array of Tensor The computation graph description of injective in the format of an array of tensors . Returns Schedule The computation schedule for the op .
def schedule_injective(outs):
    outs = [outs] if isinstance(outs, tvm.tensor.Tensor) else outs
    x = outs[0]
    s = tvm.create_schedule([x.op for x in outs])
    tvm.schedule.AutoInlineInjective(s)
    if len(s[x].op.axis) == 4:
        n, c, _, _ = s[x].op.axis
        fused = s[x].fuse(n, c) 
        s[x].parallel(fused)
    else:
        s[x].parallel(s[x].op.axis[0])
    return s
def schedule_injective ( outs ) : outs = [ outs ] if isinstance ( outs , tvm . tensor . Tensor ) else outs x = outs [ 0 ] s = tvm . create_schedule ( [ x . op for x in outs ] ) tvm . schedule . AutoInlineInjective ( s ) if len ( s [ x ] . op . axis ) == 4 : n , c , _ , _ = s [ x ] . op . axis fused = s [ x ] . fuse ( n , c ) s [ x ] . parallel ( fused ) else : s [ x ] . parallel ( s [ x ] . op . axis [ 0 ] ) return s
X86 schedule for injective op.
X86 schedule for injective op .
"""X86 schedule for injective op.

    Parameters
    ----------
    outs: Array of Tensor
          The computation graph description of injective in the format
          of an array of tensors.

    Returns
    -------
    sch: Schedule
        The computation schedule for the op.
    """ # for nhwc layout, fuse n and h
outs