Commit e7709809 authored by gracehoney's avatar gracehoney
Browse files

Use tf_optimizer.OptimizeGraph() to reimplement create_inference_graph()... 

Use tf_optimizer.OptimizeGraph() to reimplement create_inference_graph() method, and fix a bug in GetDeviceAndAllocator() where it doesn't set the cuda_device_id even if the device is found.
parent 6f251e4a
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment