Add grappler opts and new script to compile graphs.
Usage: python CompileTFGraph.py --config config.json where a sample config.json looks like this: { "model_name":"full_kernel.pb", "input_tensors":{ "actual_input_1":"2,245,234,3", "input2":"2,245,234,3" }, "output_tensors":[ "output1", "output2" ], "scale":10, "bitlength":63, "mode":"SCI", "save_weights" : true } Do python CompileTFGraph.py --help for seeing all the options.
Showing
- Athos/CompileTFGraph.py 182 additions, 0 deletionsAthos/CompileTFGraph.py
- Athos/CompileTFGraph.sh 0 additions, 206 deletionsAthos/CompileTFGraph.sh
- Athos/CompilerScripts/compile_tf.py 165 additions, 0 deletionsAthos/CompilerScripts/compile_tf.py
- Athos/CompilerScripts/grappler.py 176 additions, 0 deletionsAthos/CompilerScripts/grappler.py
- Athos/CompilerScripts/parse_config.py 181 additions, 0 deletionsAthos/CompilerScripts/parse_config.py
- Athos/CompilerScripts/tf_graph_io.py 12 additions, 0 deletionsAthos/CompilerScripts/tf_graph_io.py
- Athos/CompilerScripts/tf_graph_trans.py 0 additions, 1 deletionAthos/CompilerScripts/tf_graph_trans.py
- Athos/TFCompiler/DumpTFMtData.py 60 additions, 28 deletionsAthos/TFCompiler/DumpTFMtData.py
- Athos/TFCompiler/ProcessTFGraph.py 10 additions, 11 deletionsAthos/TFCompiler/ProcessTFGraph.py
Loading
Please register or sign in to comment