Skip to content

Commit

Permalink
Override init_thresholds() in MHA shared functional test class
Browse files Browse the repository at this point in the history
  • Loading branch information
jhajducz committed Feb 5, 2025
1 parent 5df26a3 commit 82edc77
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -42,9 +42,9 @@ class MHABase : virtual public SnippetsTestsCommon {
void SetUp() override;
void compile_model() override;
void generate_inputs(const std::vector<ov::Shape>& targetInputStaticShapes) override;
void init_thresholds() override;
virtual std::shared_ptr<SnippetsFunctionBase> get_subgraph() const = 0;
virtual void init_params(std::vector<InputShape>& input_shapes, ov::element::Type& prc, ov::AnyMap& additional_config) = 0;
virtual void init_thresholds();

size_t m_thread_count;
std::vector<ov::element::Type> m_input_types;
Expand Down

0 comments on commit 82edc77

Please sign in to comment.