提交 7dee1ad0 编写于 作者: 刘托

Merge branch 'fix_reduce_mean_gpu' into 'master'

check keep_dims for ReduceMeanOp in GPU

See merge request !872
......@@ -56,7 +56,7 @@ MaceStatus ReduceMeanKernel<T>::Compute(
const Tensor *input,
Tensor *output) {
MACE_CHECK_NOTNULL(input);
// MACE_CHECK(keep_dims_, "reduce mean gpu only support keep dims.");
MACE_CHECK(keep_dims_, "reduce mean gpu only support keep dims.");
MACE_CHECK(input->dim_size() == 4,
"reduce mean gpu only support 4-dim input");
MACE_CHECK(axis_.size() == 2 && axis_[0] == 1 && axis_[1] == 2,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册