Skip to content

Commit c541bc1

Browse files
authored
Fix return type mismatch in choose_qparams_tensor_out
Differential Revision: D61272239 Pull Request resolved: #4711
1 parent 2780528 commit c541bc1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

kernels/quantized/cpu/op_choose_qparams.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ void choose_qparams(
149149
}
150150
} // namespace
151151

152-
std::tuple<Tensor, Tensor> choose_qparams_tensor_out(
152+
std::tuple<Tensor&, Tensor&> choose_qparams_tensor_out(
153153
const Tensor& input,
154154
int64_t quant_min,
155155
int64_t quant_max,
@@ -164,7 +164,7 @@ std::tuple<Tensor, Tensor> choose_qparams_tensor_out(
164164
return {scale_out, zero_point_out};
165165
}
166166

167-
::std::tuple<Tensor, Tensor> choose_qparams_tensor_out(
167+
::std::tuple<Tensor&, Tensor&> choose_qparams_tensor_out(
168168
RuntimeContext& context,
169169
const Tensor& input,
170170
int64_t quant_min,

0 commit comments

Comments
 (0)