Add member func named is_splited_model()
This commit is contained in:
parent
2b6d2daa6a
commit
813fe5f638
|
|
@ -176,6 +176,10 @@ public:
|
|||
|
||||
virtual bool is_stateful() const override { return m_is_stateful; }
|
||||
|
||||
virtual bool is_splited_model() const override {
|
||||
return m_model_is_splitted;
|
||||
}
|
||||
|
||||
ov::PartialShape get_graph_input_shape(const ggml_tensor * op, const ggml_tensor * input) const;
|
||||
|
||||
static void dump_cgraph(const ggml_cgraph * cgraph, std::string & filename);
|
||||
|
|
|
|||
|
|
@ -66,6 +66,8 @@ public:
|
|||
|
||||
virtual bool is_stateful() const = 0;
|
||||
|
||||
virtual bool is_splited_model() const = 0;
|
||||
|
||||
virtual int is_swa_layer(int layer) const = 0;
|
||||
};
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue