delete memory copy from linear_chain_crf_op.

add_depthwiseConv_op_gpu
caoying03 8 years ago
parent 7a68787667
commit 4020451a7b

@ -187,7 +187,7 @@ class LinearChainCRFOp : public framework::OperatorWithKernel {
const framework::ExecutionContext& ctx) const override { const framework::ExecutionContext& ctx) const override {
return framework::OpKernelType( return framework::OpKernelType(
framework::ToDataType(ctx.Input<LoDTensor>("Emission")->type()), framework::ToDataType(ctx.Input<LoDTensor>("Emission")->type()),
ctx.device_context()); platform::CPUPlace());
} }
}; };
@ -248,7 +248,7 @@ class LinearChainCRFGradOp : public framework::OperatorWithKernel {
framework::ToDataType( framework::ToDataType(
ctx.Input<LoDTensor>(framework::GradVarName("LogLikelihood")) ctx.Input<LoDTensor>(framework::GradVarName("LogLikelihood"))
->type()), ->type()),
ctx.device_context()); platform::CPUPlace());
} }
}; };

File diff suppressed because it is too large Load Diff
Loading…
Cancel
Save