Add method to check for initializer

1 files changed, 12 insertions, 0 deletions

diff --git a/onnx/model.go b/onnx/model.go
index cd1625c..74c7030 100644
--- a/onnx/model.go
+++ b/onnx/model.go
@@ -40,6 +40,18 @@ func NewModel(name string) (*Model, error) {
 	return m, nil
 }
 
+func (m *Model) HasInitializer(initializer string) bool {
+	g := m.model.GetGraph()
+
+	for _, init := range g.GetInitializer() {
+		if init.GetName() == initializer {
+			return true
+		}
+	}
+
+	return false
+}
+
 func (m *Model) ExtractInitializer(initializer string) ([]float32, []int, error) {
 	g := m.model.GetGraph()