fixed

fca3500e · Jing Zhang · 115c5bfc · fca3500e
Commit fca3500e authored Aug 18, 2021 by Jing Zhang
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

host/host_tensor/include/host_conv_wrw.hpp host/host_tensor/include/host_conv_wrw.hpp +3 -3

No files found.
--- a/host/host_tensor/include/host_conv_wrw.hpp
+++ b/host/host_tensor/include/host_conv_wrw.hpp
 #pragma once
 #include "host_tensor.hpp"
-template <typename TIn,
+template <typename TOut,
-          typename TOut,
+          typename TIn,
          typename TWei,
          typename ConvStrides,
          typename ConvDilations,
          typename InLeftPads,
          typename InRightPads>
 void host_direct_convolution_backward_weights(
-    const Tensor<TIn>& in,
    const Tensor<TOut>& out,
+    const Tensor<TIn>& in,
    Tensor<TWei>& wei,
    const ConvStrides& conv_strides,
    const ConvDilations& conv_dilations,