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 {
return framework::OpKernelType(
framework::ToDataType(ctx.Input<LoDTensor>("Emission")->type()),
ctx.device_context());
platform::CPUPlace());
}
};
@ -248,7 +248,7 @@ class LinearChainCRFGradOp : public framework::OperatorWithKernel {
framework::ToDataType(
ctx.Input<LoDTensor>(framework::GradVarName("LogLikelihood"))
->type()),
ctx.device_context());
platform::CPUPlace());
}
};

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