Skip to content

Commit b1cd9ca

Browse files
committed
[CI][Fix] Remove additional qnn.op.transpose_conv2d from docs
Fixes CI after #10077, and replaces misuse elsewhere. Change-Id: I095fc8ea2b8d268b09538832cba1f5482a73a9d9
1 parent 4b0558c commit b1cd9ca

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

docs/reference/langref/relay_op.rst

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,4 +245,3 @@ This level supports dialect operators.
245245
tvm.relay.qnn.op.simulated_dequantize
246246
tvm.relay.qnn.op.simulated_quantize
247247
tvm.relay.qnn.op.subtract
248-
tvm.relay.qnn.op.transpose_conv2d

python/tvm/relay/op/strategy/generic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,7 +482,7 @@ def conv2d_transpose_strategy(attrs, inputs, out_type, target):
482482
wrap_topi_schedule(topi.generic.schedule_conv2d_transpose_nchw),
483483
name="conv2d_transpose_nchw.generic",
484484
)
485-
else: # group_transpose_conv2d
485+
else: # group_conv2d_transpose
486486
strategy.add_implementation(
487487
wrap_compute_conv2d_transpose(topi.nn.group_conv2d_transpose_nchw, has_groups=True),
488488
wrap_topi_schedule(topi.generic.schedule_group_conv2d_transpose_nchw),

python/tvm/relay/qnn/op/qnn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -527,8 +527,8 @@ def conv2d_transpose(
527527
kernel_scale: tvm.relay.Expr
528528
The scale for the weight tensor. The scale for the weight tensor is
529529
stored for access to this during relay. This information is not
530-
needed in the pass pipeline after qnn.transpose_conv2d is lowered to the
531-
sequence of steps as in nn.transpose_conv2d. See also input_scale in Requantize.
530+
needed in the pass pipeline after qnn.conv2d_transpose is lowered to the
531+
sequence of steps as in nn.conv2d_transpose. See also input_scale in Requantize.
532532
533533
strides : Tuple[int], optional
534534
The strides of convolution.

0 commit comments

Comments
 (0)