未验证 提交 77b7765f 编写于 作者: R risemeup1 提交者: GitHub

fix gcc12 error (#52318)

上级 710c13ed
......@@ -1299,7 +1299,7 @@ void batch_norm_grad(const Tensor& x,
std::vector<int> nchw_to_nhwc_dim = {0, 2, 3, 1};
std::vector<int> nhwc_to_nchw_dim = {0, 3, 1, 2};
auto reduce_axis = IntArray(std::vector<int>{0, 1, 2});
auto reduce_axis = IntArray(std::vector<int64_t>{0, 1, 2});
auto dtype = x_data.dtype();
switch (data_layout_) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册