Skip to content

Commit

Permalink
Fix comments on PR
Browse files Browse the repository at this point in the history
  • Loading branch information
guberti committed Sep 1, 2022
1 parent 1b797ab commit ebe529b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
# under the License.
"""This is a special intrinsic used for depthwise convolution using Cortex-M DSP instructions
(v7e-m). It takes as inputs an int8 HWC data tensor and an int8 CHWc kernel. This intrinsic "lays"
the kernel on top of the data tensors tarting from a given pointer, performs signed sixteen-bit
the kernel on top of the data tensors starting from a given pointer, performs signed sixteen-bit
multiplies on each pair of values, and sums all the products in an int32 accumlator. This process is
repeated four times giving four int32 outputs - one per channel."""

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -148,7 +148,7 @@ class TestDepthwiseConv2d_NHWC_HWOI(BasicDepthwiseConv2dTests):


class TestDepthwiseConv2d_NHWC_HWOI_DSP(BasicDepthwiseConv2dTests):
"""This test is for depthwise_conv2d_nhwc.generic schedule."""
"""This test is for depthwise_conv2d_nhwc_dsp.arm_cpu schedule."""

data_shape, kernel_size, num_filter, strides, padding, dilation = tvm.testing.parameters(
# The LLVM implementation doesn't support "SAME" and "VALID" padding,
Expand Down

0 comments on commit ebe529b

Please sign in to comment.