diff options
| -rw-r--r-- | onnx/model.go | 12 |
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() |