IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] optimized out detection_out if nms_threshold is under -1. (#12730)

Browse Source
This commit is contained in:
Sungeun Kim 2022-08-25 11:05:38 +09:00 committed by GitHub
parent 11a869d4d1
commit dc7f8249a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/plugins/intel_gpu/src/graph/impls/ocl/detection_output.cpp
View File

@ -50,6 +50,13 @@ private:
detectOutParams.conf_padding_y = arg.confidence().get_output_layout().data_padding.lower_size().spatial[1]; detectOutParams.conf_padding_y = arg.confidence().get_output_layout().data_padding.lower_size().spatial[1];
} }
protected:
bool optimized_out(detection_output_inst& instance) const override {
/// purpose: To optimize out detection_output for perf measurement.
/// how-to: update nms_threshold to '-100' from ir file.
return (instance.argument.nms_threshold < -1);
}
public: public:
static primitive_impl* create(const detection_output_node& arg, const kernel_impl_params& impl_param) { static primitive_impl* create(const detection_output_node& arg, const kernel_impl_params& impl_param) {
auto detect_out_params = get_default_params<kernel_selector::detection_output_params>(impl_param); auto detect_out_params = get_default_params<kernel_selector::detection_output_params>(impl_param);