From e60d6c0c488585b9508f7ac517e981c61f264cf6 Mon Sep 17 00:00:00 2001 From: Jonas Knobloch Date: Tue, 14 Jul 2026 19:06:40 +0200 Subject: Add method to check for initializer --- onnx/model.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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() -- cgit v1.2.3