Merge pull request #7675 from lcy-seso/remove_copy_from_crf

delete memory copy from linear_chain_crf_op.
add_depthwiseConv_op_gpu
Cao Ying 7 years ago committed by GitHub
commit 6720e67fbd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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