Skip to content

Commit 5e73b1e

Browse files
committed
serialize attributes
1 parent a8e776b commit 5e73b1e

File tree

1 file changed

+14
-0
lines changed
  • src/plugins/intel_cpu/src/transformations/cpu_opset/x64/op

1 file changed

+14
-0
lines changed

src/plugins/intel_cpu/src/transformations/cpu_opset/x64/op/qkv_proj.cpp

+14
Original file line numberDiff line numberDiff line change
@@ -37,4 +37,18 @@ std::shared_ptr<Node> QKVProjectionNode::clone_with_new_inputs(const ov::OutputV
3737
check_new_args_count(this, new_args);
3838
return std::make_shared<QKVProjectionNode>(new_args, m_config);
3939
}
40+
41+
bool QKVProjectionNode::visit_attributes(ov::AttributeVisitor& visitor) {
42+
INTERNAL_OP_SCOPE(QKVProjectionNode_visit_attributes);
43+
visitor.start_structure("config");
44+
visitor.on_attribute("quantized", m_config.quantized);
45+
visitor.on_attribute("hidden_size", m_config.hidden_size);
46+
visitor.on_attribute("proj_size0", m_config.proj_size0);
47+
visitor.on_attribute("proj_size1", m_config.proj_size1);
48+
visitor.on_attribute("proj_size2", m_config.proj_size2);
49+
visitor.on_attribute("weights_combined", m_config.weights_combined);
50+
visitor.finish_structure();
51+
return true;
52+
}
53+
4054
} // namespace ov::intel_cpu

0 commit comments

Comments
 (0)