017ff94d15df85ea8476f8ad3ce234470072ae84,python/tvm/relay/op/nn/_nn.py,,,#,570

Before Change

    return [topi.nn.upsampling(inputs[0], scale_h, scale_w, layout, method, align_corners)]


reg.register_injective_schedule("nn.upsampling")


// upsampling3d
@reg.register_compute("nn.upsampling3d")

After Change

    return [topi.nn.lrn(inputs[0], attrs.size, attrs.axis, attrs.alpha, attrs.beta, attrs.bias)]


reg.register_schedule("nn.lrn", strategy.schedule_lrn)
reg.register_pattern("nn.lrn", OpPattern.OPAQUE)


// upsampling
@reg.register_compute("nn.upsampling")

Instances