Reference fix

25e4687b · Bartlomiej Kocot · dcdaeb2e · 25e4687b
Commit 25e4687b authored Nov 08, 2023 by Bartlomiej Kocot
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

library/include/ck/library/reference_tensor_operation/cpu/reference_conv_fwd.hpp ...ary/reference_tensor_operation/cpu/reference_conv_fwd.hpp +2 -2

No files found.
--- a/library/include/ck/library/reference_tensor_operation/cpu/reference_conv_fwd.hpp
+++ b/library/include/ck/library/reference_tensor_operation/cpu/reference_conv_fwd.hpp
@@ -421,8 +421,8 @@ struct ReferenceConvFwd : public device::BaseOperator
        InElementwiseOperation in_element_op,
        WeiElementwiseOperation wei_element_op,
        OutElementwiseOperation out_element_op,
-        const std::array<Tensor<OutDataType>, NumAElementwiseTensor>& elementwise_a_tensors = {},
+        const std::array<Tensor<InDataType>, NumAElementwiseTensor>& elementwise_a_tensors  = {},
-        const std::array<Tensor<OutDataType>, NumBElementwiseTensor>& elementwise_b_tensors = {},
+        const std::array<Tensor<WeiDataType>, NumBElementwiseTensor>& elementwise_b_tensors = {},
        const std::array<Tensor<OutDataType>, NumDElementwiseTensor>& elementwise_d_tensors = {})
    {
        return Argument{input,