From e21fa76fa513c6942210c076746fa3764454fe54 Mon Sep 17 00:00:00 2001 From: xiebaiyuan Date: Thu, 14 Mar 2019 16:37:10 +0800 Subject: [PATCH] remove unimpled gpu op in load --- src/framework/load_ops.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/framework/load_ops.h b/src/framework/load_ops.h index d7e2a059ed..a969a73b9e 100644 --- a/src/framework/load_ops.h +++ b/src/framework/load_ops.h @@ -65,7 +65,7 @@ limitations under the License. */ LOAD_OP(feed) LOAD_OP(fetch) #ifdef FILL_CONSTANT_OP -LOAD_OP(fill_constant) +LOAD_OP2(fill_constant, CPU, FPGA) #endif #ifdef BATCHNORM_OP LOAD_OP2(batch_norm, CPU, GPU_CL); @@ -185,7 +185,7 @@ LOAD_OP1(crf_decoding, CPU); LOAD_OP2(mul, CPU, GPU_CL); #endif #ifdef NORM_OP -LOAD_OP2(norm, CPU, GPU_CL); +LOAD_OP1(norm, CPU); #endif #ifdef RELU_OP LOAD_OP2(relu, CPU, GPU_CL); -- GitLab