ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onnx-ops.py
Go to the documentation of this file.
1#!/usr/bin/env python3
2
3# Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17import onnx
18import os
19import sys
20
21
22def _dump_operators(onnx_model):
23 for node in onnx_model.graph.node:
24 print(node.op_type)
25
26
27def _help_exit(cmd_name):
28 print('Dump ONNX model file Operators')
29 print('Usage: {0} [onnx_path]'.format(cmd_name))
30 print('')
31 exit()
32
33
34def main():
35 if len(sys.argv) < 2:
36 _help_exit(os.path.basename(sys.argv[0]))
37
38 onnx.checker.check_model(sys.argv[1])
39 onnx_model = onnx.load(sys.argv[1])
40
41 _dump_operators(onnx_model)
42
43
44if __name__ == "__main__":
45 main()
_dump_operators(onnx_model)
Definition onnx-ops.py:22
_help_exit(cmd_name)
Definition onnx-ops.py:27