search

tensorflow / ngraph-bridge

TensorFlow-nGraph bridge
Other
137 stars 64 forks source link

Enable shape information for nGraph graphs #730

Closed kanvi-nervana closed 3 years ago

kanvi-nervana commented 3 years ago

Enable shape info for nGraph graphs when dumping those by setting the env variables in the code "NGRAPH_VISUALIZE_TREE_OUTPUT_SHAPES" "NGRAPH_VISUALIZE_TREE_OUTPUT_TYPES" "NGRAPH_VISUALIZE_TREE_IO"