Skip to content

Why are Some ONNX Model Layers Ignored by Profiler ? #77

Discussion options

You must be logged in to vote

To print all layers 's profile details

1 Clone repo

git clone https://github.com/ThanatosShinji/onnx-tool.git

2 Make changes in the init.py.

# These ops have no computation
# NoMacsOps = (
#     'Identity', 'Constant', 'Shape', 'Squeeze', 'Unsqueeze', 'Reshape', 'ConstantOfShape', 'Cast', 'Pad', 'Concat',
#     'Slice', 'Gather'
# )

#--------------------------------Change to empty
NoMacsOps = ()

3 Install

pip install <local_dir>/onnx-tool/

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by markreeves225
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
1 participant