From 052c5de51e57bf1717bd71ce49d6f7fcc4a346b7 Mon Sep 17 00:00:00 2001 From: yejianwu Date: Thu, 15 Nov 2018 14:45:50 +0800 Subject: [PATCH] check keep_dims for ReduceMeanOp in GPU --- mace/ops/opencl/image/reduce_mean.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/ops/opencl/image/reduce_mean.h b/mace/ops/opencl/image/reduce_mean.h index ca5daa5f..78c2e7e5 100644 --- a/mace/ops/opencl/image/reduce_mean.h +++ b/mace/ops/opencl/image/reduce_mean.h @@ -56,7 +56,7 @@ MaceStatus ReduceMeanKernel::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, -- GitLab