[GPU] optimized out detection_out if nms_threshold is under -1. (#12730)
This commit is contained in:
parent
11a869d4d1
commit
dc7f8249a7
@ -50,6 +50,13 @@ private:
|
||||
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:
|
||||
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);
|
||||
|
Loading…
Reference in New Issue
Block a user