3734d5f7f8475a2a7897f239b9942c913256fc96,python/tvm/relay/op/nn/_nn.py,,,#,570

Before Change

    "nn.contrib_conv2d_gemm_without_weight_transform",
    strategy.conv2d_gemm_without_weight_transform_strategy,
)
reg.register_pattern(
    "nn.contrib_conv2d_gemm_without_weight_transform", OpPattern.OUT_ELEMWISE_FUSABLE
)


@reg.register_compute("nn.contrib_conv2d_gemm_weight_transform")
def compute_contrib_conv2d_gemm_weight_transform(attrs, inputs, out_dtype):

After Change

    ]


reg.register_injective_schedule("nn.upsampling3d")


// pad
reg.register_broadcast_schedule("nn.pad")

Instances