diff --git a/src/framework/load_ops.h b/src/framework/load_ops.h index d7e2a059ed3fc13e0ec650db872cd6d1d0013d04..a969a73b9edfe1e7a475ea7191bfcd552d09357c 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);